numpy.roll#

numpy.roll(a, shift, axis=None)[源代码]#

沿给定轴滚动数组元素。

滚动到最后一个位置之外的元素将在第一个位置重新引入。

参数:

a类数组对象: 输入数组。
shiftint 或 int 元组: 元素被移位的次数。如果为元组，则 axis 必须是相同大小的元组，并且给定的每个轴都由相应数量进行移位。如果为 int，而 axis 是 int 的元组，则对所有给定的轴使用相同的值。
axisint 或 int 元组，可选: 元素被移位的轴。默认情况下，在移位之前先展平数组，然后恢复原始形状。

返回:

resndarray: 输出数组，形状与 a 相同。

另请参阅

rollaxis: 向后滚动指定的轴，直到它位于给定的位置。

备注

支持同时在多个维度上滚动。

示例

>>> import numpy as np
>>> x = np.arange(10)
>>> np.roll(x, 2)
array([8, 9, 0, 1, 2, 3, 4, 5, 6, 7])
>>> np.roll(x, -2)
array([2, 3, 4, 5, 6, 7, 8, 9, 0, 1])

>>> x2 = np.reshape(x, (2, 5))
>>> x2
array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])
>>> np.roll(x2, 1)
array([[9, 0, 1, 2, 3],
       [4, 5, 6, 7, 8]])
>>> np.roll(x2, -1)
array([[1, 2, 3, 4, 5],
       [6, 7, 8, 9, 0]])
>>> np.roll(x2, 1, axis=0)
array([[5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4]])
>>> np.roll(x2, -1, axis=0)
array([[5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4]])
>>> np.roll(x2, 1, axis=1)
array([[4, 0, 1, 2, 3],
       [9, 5, 6, 7, 8]])
>>> np.roll(x2, -1, axis=1)
array([[1, 2, 3, 4, 0],
       [6, 7, 8, 9, 5]])
>>> np.roll(x2, (1, 1), axis=(1, 0))
array([[9, 5, 6, 7, 8],
       [4, 0, 1, 2, 3]])
>>> np.roll(x2, (2, 1), axis=(1, 0))
array([[8, 9, 5, 6, 7],
       [3, 4, 0, 1, 2]])