标题可能有点糊涂,我在这里解释一下:给定一个数组:
arr = np.array([1.0, 2.0])
并给出一个索引列表:
idxes = np.array([0, 0, 1, 0])
以及要添加的号码列表:
e = np.array([0.1, 0.35, 0.5, 0.2])
我想要的是得到这样的结果:
arr[idxes] += e
意思是
[arr[0], arr[0], arr[1], arr[0]] + [0.1, 0.35, 0.5, 0.2]
使arr[0]
加三倍arr[0] = arr[0] + 0.1 + 0.35 + 0.2 = 1.65
,并arr[1]
添加一个时间arr[1] = arr[1] + 0.5 = 2.5
,
我怎么能做到这一点的情况下直接for循环?
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句