轮转就是把数组向右移动几个,然后多出来的部分回到起点。
比如
[1, 2, 3, 4]
向右轮转1格
[4, 1, 2, 3]
那么我们可以使用np.roll来实现
>>> a = np.array([1,2,3,4])
>>> b = np.roll(a, shift=1,axis=0)
>>> b
array([4, 1, 2, 3])
shift表示移动几个格子,axis表示在哪个轴上移动(由于我们是一维数组,所以不用管它)
我们其实可以将其改写为普通的for循环。这样,我们可以把代码迁移到比如c++上去。
import numpy as np
a = np.array([1,2,3,4])
# b = np.roll(a,1)
b = np.zeros(shape=a.shape,dtype=int)
for i in range(a.shape[0]):
ind=(i - 1) % a.shape[0]
print(ind)
b[i] = a[(i - 1) % a.shape[0]]
output
3
0
1
2
array([4, 1, 2, 3])
其实,我们就是观察下标的变化
可以得知
a的第0号元素在b中为第3号
a的第1号元素在b中为第1号
a的第2号元素在b中为第2号
a的第3号元素在b中为第3号