概要
MNISTは言わずと知れた文字画像セットですね。それを使う上で「1」だけ喉から手が出るほどほしい！！！となることがあるかと思います(大袈裟)。そんなときの小技を紹介します。
方法
import keras
(x_train, y_train), (x_test, y_test) = keras.datasets.mnist.load_data()
char = 1 #0でもなんでもOK
x_train_1 = x_train[np.where(y_train==char)]
↑こんだけ。
最後に
(小技でもなんでもない)