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号

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值