Write plot image using Matplotlib pyplot from scratch

single image

multiple image

import matplotlib.pyplot as plt
from keras.datasets import mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data()

sample = 1
image = x_train[sample]
label = y_train[sample]
fig = plt.figure(1)
plt.imshow(image, cmap='gray_r')
plt.show()

num = 10
images = x_train[:num]
labels = y_train[:num]

rows, cols = 2, 5

fig, axes = plt.subplots(rows, cols, figsize = (1.5*cols, 2* rows))
for i in range(num):
    ax = axes[i//cols, i%cols]
    ax.imshow(images[i], cmap='gray')
    ax.set_title('Label: {}'.format(labels[i]))
plt.tight_layout()
plt.show()