温馨提示×

NumPy中怎么循环滚动数组

小亿
126
2024-05-11 18:29:50
栏目: 编程语言

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

0