numpy.roll

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

Сдвиг элементов массива вдоль заданной оси.

Элементы, которые сдвигаются за пределы последней позиции, возвращаются в начало.

Параметры:	a : array_like Входной массив. shift : int Число позиций, на которые сдвигаются элементы. axis : int, необязательно Ось, вдоль которой происходит сдвиг. По умолчанию массив сглаживается перед сдвигом, после чего восстанавливается исходная форма.
Возвращает:	res : ndarray Выходной массив, имеющий ту же форму, что и a.

См. также

rollaxis

Сдвигает указанную ось назад, пока она не окажется в заданной позиции.

Примеры

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

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