numpy.roll

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

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

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

Параметры:

aarray_like

Входной массив.

shiftint или кортеж из целых чисел

Количество позиций, на которые сдвигаются элементы. Если кортеж, то axis должен быть кортежем той же длины, и каждая из заданных осей сдвигается на соответствующее число. Если целое число, а axis — кортеж из целых чисел, то одно и то же значение используется для всех заданных осей.

axisint или кортеж из целых чисел, необязательно

Ось или оси, вдоль которых сдвигаются элементы. По умолчанию массив сжимается в одномерный массив, затем сдвигается, после чего восстанавливается исходная форма.

Возвращает:

resndarray

Выходной массив с такой же формой, что и 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]])
>>> 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]])