numpy.rollaxis函数解析

numpy.rollaxis函数解析

 

官方文档里写的非常清楚

 

numpy.rollaxis(a, axis, start=0)

a是输入的数组

axis是想要滚动的轴

start是滚动轴的目标位置，当start>axis时，axis轴滚动到start-1处，当start<=axis时，axis滚动到start处（start的默认值为0）

例如：

>>> a = np.ones((3,4,5,6))
#数组a=[3,4,5,6]，位置从0开始计数
>>> np.rollaxis(a, 3, 1).shape
(3, 6, 4, 5)
#axis=3，start=1，start<axis，故需要将a中的第三列，即6，滚回到位置为1处，数组变为[3,6,4,5]

>>> np.rollaxis(a, 2).shape
(5, 3, 4, 6)
#axis=2,start=0, start<axis，故需要将a中的第二列，即5，滚回到位置为0处，数组变为[5,3,4,6]

>>> np.rollaxis(a, 1, 4).shape
(3, 5, 6, 4)
#axis=1,start=4, start>axis, 故需要将a中的第一列，即4，滚动到位置为3处，数组变为[3,5,6,4]