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