顯示來自一個numpy陣列的多個圖像

Question

我有一個numpy數組（data.npy），它存儲了多個示例圖像。我想查看/保存所有圖像。我曾嘗試以下操作：

img_array=np.load('data.npy') 
i = 0 
while i < len(img_array): 
    plt.imshow(img_array[i], cmap='gray') 
    plt.show() 
    i += 1 

但是這給出了一個錯誤：

TypeError: Invalid dimensions for image data 

Answer 1

根據@John Zwinck的回答，以下代碼似乎對我來說工作得很好。

In [12]: for idx, el in enumerate(img_array): 
    ...:  plt.imshow(np.moveaxis(img_array[idx], 0, -1), cmap='gray') 
    ...: 

np.moveaxis圍繞陣列的軸移動。在這裏它將原始數組中的第一個軸移動到最後一個軸。

In [13]: a[10].shape 
Out[13]: (4, 100, 100) 

In [14]: np.moveaxis(a[10], 0, -1).shape 
Out[14]: (100, 100, 4) 

Answer 2

你傳遞一個(4,100,100)陣列imshow()但該文檔說，它要求渠道維度放在最後：http://matplotlib.org/api/pyplot_api.html#matplotlib.pyplot.imshow

所以使用moveaxis()或rollaxis()更改尺寸的順序：https://docs.scipy.org/doc/numpy/reference/generated/numpy.moveaxis.html