移調3-d numpy的陣列

Question

假設我有一個numpy陣列A具有形狀(5,24,1)。
我想利用五個單獨的轉置沿axis = 0，使得轉置後所得到的形狀是(5,1,24)。
我怎樣才能做到這一點使用一些numpy功能？

Answer 1

三種方法可以建議 -

A.swapaxes(1,2) 
A.transpose(0,2,1) 
np.rollaxis(A,2,1) 

對於swapaxes和transpose，請按照他們的文檔。

隨着np.rollaxis，我們正在滾動的軸，以使第三軸到第二位置，從而推回第二軸線成第三軸的位置。鏈接的文檔中有更多信息。

使用的事實，最後軸是單 -

A[:,None,:,0] 

Answer 2

你可以使用np.moveaxis（一個選項從Divakars出色答卷丟失），您可以定義哪些軸應移動：

np.moveaxis(A, (0, 1, 2), (0, 2, 1)) 

這將軸0移動到0（無變化，也可以省略），1至2和2至1。因此，這基本上只是交換軸1和2.

>>> A = np.ones((5,24,1)) 
>>> res = np.moveaxis(A, (1, 2), (2, 1)) # this time without the zeros 
>>> res.shape 
(5, 1, 24)