如何在numpy中向前滚动轴？

Question

numpy.rollaxis向后滚动轴。假设我的ndarray形状为(3,640,480)，我想将第一个轴滚动到最后一个轴，将形状更改为(640,480,3)。我能轻易做到吗？虽然在这个简单的情况下，numpy.rollaxis的两个调用都可以工作，如果我有n轴，这将是笨拙的。

Answer 1

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

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