Python numpy.roll函数方法的使用-CJavaPy

numpy.roll

numpy.roll(a, shift, axis=None) [source]

沿给定轴滚动数组元素。

超出最后位置的元素将在第一个位置重新引入。

参数：

a ：array_like

输入数组。

shift ：int 或 int类型的tuple

元素移位的位置数。如果是一个元组，

那么轴必须是一个相同大小的元组，

并且每个给定的轴被相应的数字移动。如果值是int 当axis是int类型的tuple，

那么对所有给定的轴使用相同的值。

axis ：int 或 int类型的tuple, 可选

元素沿其移动的一个或多个轴。默认情况下，

在移动之前将阵列弄平，然后恢复原始形状。

返回值：

res ：ndarray

输出数组，shape与a相同。

Notes

1.12.0版中的新功能。

支持同时滚动多个维度。

例子

1）一维数组简单循环

import numpy as np

a = np.array([0, 1, 2, 3, 4, 5])
print(np.roll(a, 2))

2）二维数组按行循环

import numpy as np

b = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])
 # 沿着第0轴（行）下移1位
print(np.roll(b, shift=1, axis=0))

3）二维数组按列循环

import numpy as np

b = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])

# 沿着第1轴（列）左移1位
print(np.roll(b, shift=-1, axis=1))

4）不指定轴，直接滚动（相当于扁平化后滚动）

import numpy as np

b = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])

print(np.roll(b, shift=3))