将第一根轴放在最后一根轴上的最快方法是什么? [numpy的]

时间:2016-09-20 17:22:28

标签: python numpy

让我们看看这个带有N0 x N1 x ... x Nm维数组的愚蠢例子,例如

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)。

有什么好主意吗?

1 个答案:

答案 0 :(得分:1)

  

我尝试使用np.rollaxis,但这似乎是错误的功能。

不,这是正确的功能。

x = numpy.rollaxis(x, 0, x.ndim)