실행 결과

학습셋 이미지 수: 60000개
테스트셋 이미지 수: 10000개

불러온 이미지 중 한 개만 다시 불러와 보겠습니다. 이를 위해 먼저 맷플롯립 라이브러리를 불러옵니다. 그리고 imshow() 함수를 이용해 이미지를 출력할 수 있습니다. 모든 이미지가 X_train에 저장되어 있으므로 X_train[0]으로 첫 번째 이미지를, cmap='Greys' 옵션을 지정해 흑백으로 출력되게 합니다.

import matplotlib.pyplot as plt

plt.imshow(X_train[0], cmap='Greys')
plt.show()

실행하면 그림 16-2와 같은 이미지가 출력됩니다.

그림 16-2 | MNIST 손글씨 데이터의 첫 번째 이미지