cifar10データセットから特定のラベルを取り出したい

Question

cifar10に限る話ではないと思いますが,イメージが湧きやすいのと現在使用してるデータセットがcifar10であることからこのような質問をさせて頂きました.
下記の様にデータセットを読み込んだのちに,Y_trainが0であるX_trainとY_trainのペアを取り出して新たなデータセットを作成するにはどのようにすればよろしいでしょうか.出来るだけ,簡潔なコードでお願いしたいです.よろしくお願いします.

# Loading data
from keras.datasets import cifar10

(X_train, Y_train), (X_test, Y_test) = cifar10.load_data()

print('X_train shape: {0}, Y_train shape: {1}'.format(X_train.shape, Y_train.shape))

X_train shape: (50000, 32, 32, 3), Y_train shape: (50000, 1)

試したこと

idx = np.where(Y_train == 7)

X_train = X_train[idx]
Y_train = Y_train[idx]

Answer 1

test_data = np.delete(test_data, np.where(test_labels == 9)[0], axis=0)