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

Question

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

我尝试过：

a.swapaxis(0,-1)

它不仅将零轴放到最后，还将最后一个轴放到开头。

我尝试使用rollaxis但是不理解它的作用。

- Dims

---

你想移动它并让其他的不受影响吗？ - wwii

如果 a.shape 是 (2,3,4,5)，最终的形状是什么？ - wwii

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

- Andy Hayden