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

How to move zero axis to the last position in python?

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

我试过了

a.swapaxis(0,-1)

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

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

我想你在找 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]]])