Python numpy roll用法及代碼示例

本文簡要介紹 python 語言中 numpy.roll 的用法。

用法:

numpy.roll(a, shift, axis=None)

沿給定軸滾動數組元素。

超出最後一個位置的元素首先重新引入。

參數：

a： array_like

輸入數組。

shift： int 或整數元組

元素移動的位置數。如果是元組，則軸必須是相同大小的元組，並且每個給定軸都移動相應的數字。如果int while axis 是一個整數元組，則相同的值將用於所有給定的軸。

axis： int 或整數元組，可選

元素沿其移動的一個或多個軸。默認情況下，數組在移動之前會被展平，之後會恢複原始形狀。

返回：

res： ndarray

輸出數組，形狀與a.

注意：

支持同時滾動多個維度。

例子：

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