绘制来自 TensorFlow 数据集的数据图像，仅显示一张图像

Question

我在 TensorFlow 上处理时尚 MNIST 数据集，我试图绘制具有特定 test_label 的 train_data 的图像。我运行以下代码有效，但它只显示一张图片，即使有很多这样的图片。

for i in range (len(test_data)):
  if test_labels[i]==9:
    plt.imshow(test_data[i])

以下是我得到的输出：

Answer 1

每次迭代测试数据时，它都会覆盖之前的绘图。

如果你想得到不同的情节，使用这样的东西：

j = 0
n_rows = 2
n_cols = 3
for i in range(len(test_data)):
    if test_labels[i]==9:
        j += 1
        ax = plt.subplot(n_rows,n_cols,j)
        ax.imshow(test_data[i])
    if j >= (n_rows*n_cols):
        break

结果：

绘制来自 TensorFlow 数据集的数据图像，仅显示一张图像

Plotting images of data from TensorFlow datasets, shows one image only

python-3.x

tensorflow

tensorflow-datasets