在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函数,可以方便地实现数组的循环滚动操作。