有没有办法只导入带有0和1的MNIST图像？

import numpy as np
from tensorflow.keras.datasets import mnist

# 加载MNIST数据集
(x_train, y_train), (x_test, y_test) = mnist.load_data()

# 将图像数据转换为0和1的格式
x_train_binary = np.where(x_train > 0, 1, 0)
x_test_binary = np.where(x_test > 0, 1, 0)