我有一个形状为[5000, 255,255,1]的数组。 5000表示图像的数量; （255,255）是图像的大小
大多数条目值为0，而某些条目值为1.我如何知道这些条目的索引或位置值为1.
最后，如果事实证明[100，：，：1]有一些条目1.我想将此特定图像/矩阵显示为热图。怎么做？
答案 0 :(得分：0)
我认为你想要的是np.where，当你为它提供一个值为0或非零的数组时，它将为你提供所有非零的索引一个n维的输入。因此，该代码将为您提供一组包含非零像素的所有图像：
i,x,y,_=np.where(arr)
print(set(i))
然后您可以显示您想要的图像：
plt.imshow(arr[list(set(i))[0],:,:,0],cmap='hot')
plt.show()