现在再说rollaxis函数,这个函数有三个参数
numpy.rollaxis(arr, axis, start)
arr:输入数组
axis:要向后滚动的轴,其它轴的相对位置不会改变
start:默认为零,表示完整的滚动。会滚动到特定位置
import numpy as np
a = np.arange(8).reshape(2, 2, 2)
print '原数组:'
print a
print '\n'
# 将轴 2 滚动到轴 0(宽度到深度)
print '调用 rollaxis 函数:'
print np.rollaxis(a, 2)
# 将轴 0 滚动到轴 1:(宽度到高度)
print '\n'
print '调用 rollaxis 函数:'
print np.rollaxis(a, 2, 1)
原数组:
[[[0 1]
[2 3]]
[[4 5]
[6 7]]]
调用 rollaxis 函数:
[[[0 2]
[4 6]]
[[1 3]
[5 7]]]
调用 rollaxis 函数:
[[[0 2]
[1 3]]
[[4 6]
[5 7]]]
程序运行np.rollaxis(a, 2)时,讲轴2滚动到了轴0前面,其他轴相对2轴位置不变(start默认0),数组下标排序由0,1,2变成了1,2,0
这时数组按下标顺序重组,例如第一个数组中[0,1]下标为[000,001],其中0的下标变动不影响值,1位置的下标由001变成010,第一位的下标滚动到最后一位下标的后面,值由1(001)变成2(010):
可以得出轴的滚动就是下标的滚动,同理,运行np.rollaxis(a, 2, 1)时将下标0,1,2变为0,2,
numpy.swapaxes(arr, axis1, axis2)函数则是交换,将axis1和axis2进行互换