4-D Numpy數組的互換順序

Question

我有一個尺寸為96x96x3x1000的4-D Numpy數組-這些對應於我已導入的圖像數據集：96X96像素各1000幅圖像，每個像素RGB值。

但是，我需要遍歷每個圖像的展平數組，即。 僅是二維數組[1000] [96 * 96 * 3]。 我設法通過先做來變換給定的數組

    a.reshape(-1,a.size[3])

然后使用循環將每一列分配給圖像。 我想問一下是否有一種更簡單的切片方法來交換ndarray的順序？

謝謝

Answer 1

您可以使用numpy.swapaxes更改軸的numpy.swapaxes

a.reshape(-1,1000).swapaxes(0,1)

或簡單地轉譯

a.reshape(-1,1000).T

您也可以更改軸的順序在與開始numpy.transpose ，然后應用reshape

a.transpose([3,0,1,2]).reshape(1000,-1)