numpy.roll#
- numpy.roll(a, shift, axis=None)[source]#
沿着给定轴滚动数组元素。
滚出最后一个位置的元素将在第一个位置重新引入。
- 参数::
- a类数组
输入数组。
- shift整数或整数元组
元素移动的位数。如果为元组,则 axis 必须是相同大小的元组,并且每个给定的轴都将移动相应的数字。如果为整数,而 axis 是整数元组,则对所有给定的轴使用相同的值。
- axis整数或整数元组,可选
沿其移动元素的轴或轴。默认情况下,数组在移动之前会被扁平化,之后恢复原始形状。
- 返回值::
- 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]])