matplotlibで正方に画像を描画
概要
matplotlibを使って,MNISTでニューラルネットワーク試してみました!でよくみる以下ようなplotのやり方.
コード
from keras.datasets import mnist
import numpy as np
from matplotlib import pyplot as plt
%matplotlib inline
# MNIST data(28x28 matrix)
(X_train, y_train), (X_test, y_test) = mnist.load_data()
# visualization
def draw_digit(data, row, col, n):
size = 28
plt.subplot(row, col, n)
plt.imshow(data)
plt.gray()
show_size = 10
total = 0
plt.figure(figsize=(20,20))
for i in range(show_size):
for j in range(show_size):
draw_digit(X_train[total], show_size, show_size, total+1)
total += 1
plt.show()
肝心なのは plt.subplot(row, col, n)
.
row
とcol
はそれぞれ行数と列数で,固定値.
今回は10x10なので常に10.
n
には1スタートでその行列の何番目に描画するかを指定.