dask.array.roll

dask.array.roll

dask.array.roll(array, shift, axis=None)[源代码]

沿给定轴滚动数组元素。

此文档字符串是从 numpy.roll 复制的。

Dask 版本可能存在一些不一致性。

超出最后一个位置的元素会重新出现在第一个位置。

参数
aarray_like (Dask 中不支持)

输入数组。

shiftint 或 int 的元组

元素被移动的位置数。如果是元组,则 axis 必须是相同大小的元组,并且每个给定的轴都按相应的数量移动。如果是整数而 axis 是整数的元组,则所有给定的轴都使用相同的值。

int 或 int 的元组,可选

元素沿其移动的轴或轴。默认情况下,数组在移动前会被展平,移动后会恢复原始形状。

返回
resndarray

输出数组,形状与 a 相同。

参见

rollaxis

将指定的轴向后滚动,直到它位于给定的位置。

注释

1.12.0 新版功能.

支持同时滚动多个维度。

示例

>>> 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]])