在NumPy中,可以使用roll函數來循環滾動數組。roll函數可以接受一個數組和一個整數作為參數,表示要滾動的步數。正數表示向右滾動,負數表示向左滾動。
下面是一個例子,展示如何使用roll函數循環滾動數組:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
print("原始數組:", arr)
# 向右滾動2步
rolled_arr = np.roll(arr, 2)
print("向右滾動2步:", rolled_arr)
# 向左滾動3步
rolled_arr = np.roll(arr, -3)
print("向左滾動3步:", rolled_arr)
輸出結果為:
原始數組: [1 2 3 4 5]
向右滾動2步: [4 5 1 2 3]
向左滾動3步: [4 5 1 2 3]
通過roll函數,可以方便地實現數組的循環滾動操作。