numpy.
roll
沿给定轴滚动数组元素。
滚动到最后一个位置之外的元素在第一个位置重新引入。
输入数组。
元素移动的位置数。如果是一个元组,那么 axis 必须是大小相同的元组,并且每个给定的轴都被相应的数字移动。如果一个int axis 是整数的元组,然后对所有给定的轴使用相同的值。
元素沿其移动的轴。默认情况下,数组在移动之前被展平,之后恢复原始形状。
输出数组,形状与 a .
参见
rollaxis
向后滚动指定轴,直到其位于给定位置。
笔记
1.12.0 新版功能.
支持同时在多个维度上滚动。
实例
>>> 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]])