test.py

import gzip
import numpy as np
import matplotlib.pyplot as plt


f = gzip.open('train-images-idx3-ubyte.gz','r')

image_size = 28
num_images = 5

f.read(16)
buf = f.read(image_size * image_size * num_images)
data = np.frombuffer(buf, dtype=np.uint8).astype(np.float32)
data = data.reshape(num_images, image_size, image_size, 1)

image = np.asarray(data[0]).squeeze()
plt.imshow(image)
#plt.show()

f = gzip.open('train-labels-idx1-ubyte.gz','r')
f.read(8)
count =0
while True:
    try:
        buf = f.read(1)
        labels = np.frombuffer(buf, dtype=np.uint8).astype(np.int64)
        print(labels)
        break
        if not list(labels):
            print(count)
            break
        count += 1
    except Exception:
        print(count)
        break