python - 如何将零轴移动到python中的最后一个位置？

Question

python中如何将零轴移动到最后一个位置？

我试过了

a.swapaxis(0,-1)

但它不仅把零轴放在最后，而且把最后一个轴放在开头。

我尝试使用 rollaxis 但不明白它的作用。

Answer 1

我认为您正在寻找 moveaxis :

In [11]: a = np.arange(8).reshape(2, 2, 2)

In [12]: a
Out[12]:
array([[[0, 1],
        [2, 3]],

       [[4, 5],
        [6, 7]]])

In [13]: a.swapaxes(0, -1)
Out[13]:
array([[[0, 4],
        [2, 6]],

       [[1, 5],
        [3, 7]]])

In [14]: np.moveaxis(a, 0, -1)
Out[14]:
array([[[0, 4],
        [1, 5]],

       [[2, 6],
        [3, 7]]])