numpy - 如何在 numpy 中向前滚动轴？

Question

numpy.rollaxis向后滚动轴。假设我有一个形状为 (3,640,480) 的 ndarray，我想将第一个轴滚动到最后一个轴，将形状更改为 (640,480,3)。我可以轻松做到这一点吗？尽管在这个简单的例子中，两次调用 numpy.rollaxis 都可以，但如果我有 n 轴，这将很笨拙。

Answer 1

还有一个可选的第三个参数，start，它告诉轴应该在哪里结束。在这种情况下，您想要在轴 3 之前移动轴 0:

>>> x.shape
(3,640,480)
>>> x2 = np.rollaxis(a,0,3)
>>> x2.shape
(640,480,3)