以下所有数组都是 NumPy 数组。
我有一个数字数组,比如说a = [4, 5, 6]
。我想将它们添加到一个累积数组中,比如说,s = [0, 0, 0]
但我想控制哪个数字去哪里。例如,我想要
s[1] += a[0]
,s[2] += a[1]
, 接着s[2] += a[2]
.
所以我设置了一个辅助数组i = [1, 2, 2]
,希望它能s[i] += a
起作用。但它不会;
s[2]
最终只接收a[2]
,好像s[i] += a
是由
t = [0, 0, 0]; t[i] = a; s += t
.
我想知道是否有一种方法可以实现我的“ s[i] += a
”版本,而不必在纯 python 中进行 for 循环,因为我听说后者要慢得多。