numpy.rollaxis 函数
此函数向后滚动指定的轴,直到它位于指定的位置。 该函数采用三个参数。
numpy.rollaxis(arr, axis, start)
这里,
序号 | 参数 & 描述 |
---|---|
1 | arr 输入数组 |
2 | axis 向后滚动的轴。 其他轴的位置不会相对于彼此改变 |
3 | start 默认情况下为零会导致完整滚动。 滚动直到指定位置 |
示例
# It creates 3 dimensional ndarray import numpy as np a = np.arange(8).reshape(2,2,2) print 'The original array:' print a print '\n' # to roll axis-2 to axis-0 (along width to along depth) print 'After applying rollaxis function:' print np.rollaxis(a,2) # to roll axis 0 to 1 (along width to height) print '\n' print 'After applying rollaxis function:' print np.rollaxis(a,2,1)
它的输出结果如下 −
The original array: [[[0 1] [2 3]] [[4 5] [6 7]]] After applying rollaxis function: [[[0 2] [4 6]] [[1 3] [5 7]]] After applying rollaxis function: [[[0 2] [1 3]] [[4 6] [5 7]]]