Проблема с итерацией по массиву для построения 20 изображений из набора данных с использованием списка массивов в качестве индексов

Question

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

Я пытаюсь построить изображения из xtrain, используя plt.imshow(xtrain[rand20[i]]), где в идеале я перебираю массив 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])