2016-09-20 51 views
1

讓我們來看看有N0 X N1 X ... X牛維數組,這個愚蠢的例子,如將第一個軸放在最後一個軸後面的最快方法是什麼? [numpy的]

import numpy as np 
x = np.random.random([N0,N1,...,Nm]) 

現在我想的N0尺寸爲後面的最後一個,即

np.swapaxis np.swapaxis(np.swapaxis(x, 0,1),1,2) ... # swapping m times 

我試過np.rollaxis,但這似乎是錯誤的功能。最後,我想要一個有形狀的陣列(N1,N2,N3,...,Nm,N0)。

任何好主意?

+2

試'arr.transpose(範圍(1,NM)+ [0])' – gobrewers14

+0

「我試圖與np.rollaxis,但這似乎是錯誤函數」。 - 你是如何得出這是錯誤的功能的結論? – user2357112

回答

1

我試過np.rollaxis,但這似乎是錯誤的功能。

不,這是正確的功能。

x = numpy.rollaxis(x, 0, x.ndim) 
相關問題