循环显示图像并根据图片添加标签

Question

我有一个对应于图片的numpy数组列表。我想给这些照片贴标签。为此，我创建了一个循环，在这个循环中，我显示每个图片，并使用input()函数请求标签

问题是窗口仅在我输入标签后显示图片

以下代码复制了我的问题：

import numpy as np
import matplotlib.pyplot as plt

data = [np.random.rand(32,22), np.random.rand(32,32)]

for i in range(len(data)):
    plt.imshow(data[i])
    plt.show()
    label = input('Enter label\n')