让我们看一下这个愚蠢的例子,它有一个 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) 的数组。
有什么好的想法吗?
最佳答案
I tried with np.rollaxis, but this seems to be the wrong function.
不,这是正确的函数。
x = numpy.rollaxis(x, 0, x.ndim)
关于python - 将第一个轴放在最后一个轴后面的最快方法是什么? [numpy],我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39600233/