使用数组列表作为索引从数据集中迭代数组以绘制 20 张图像时出现问题

Question

rand20 = array([ 80, 337, 275, 651, 226, 
681, 282, 958, 150, 790, 
954, 929, 846, 848, 177, 
165, 280, 196, 694, 159])

我正在尝试使用plt.imshow(xtrain[rand20[i]])从 xtrain 绘制图像，理想情况下我遍历数组 rand20。

我还需要创建一个 5x4 子图

plt.subplot(5,4,1) 
plt.imshow(xtrain[rand20[0]])

我被困在这里，任何帮助将不胜感激。

Answer 1

您可以在此处使用zip：

fig, axes = plt.subplots(5,4,1)
for idx, ax in zip(rand20, axes.ravel()):
ax.imshow(xtrain[idx])