%matplotlib inline import numpy as np import matplotlib.pyplot as plt from PIL import Image img = Image.open('small_school.png') img_data = np.array(img) img_data.shape type(img_data) plt.imshow(img_data, cmap='gray', interpolation='nearest') img_data orig_shape = img_data.shape n_pixels = np.prod(orig_shape) n_pixels 24 * 30 flat_image = img_data.reshape(n_pixels) flat_image.shape img_2d_again = flat_image.reshape(orig_shape) plt.imshow(img_2d_again, cmap='gray', interpolation='nearest')