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]