np.roll轮转改为简单的for循环

轮转就是把数组向右移动几个，然后多出来的部分回到起点。

比如

[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号