在Numpy中有效地将几个值汇总到几个箱中

时间:2016-10-21 11:00:33

标签: python numpy

我有以下脚本:

values = np.array([1, 1, 1, 1, 1, 1]) # ndarray with some values, of length n = 6
target = np.array([1, 1, 0, 2, 3, 2], dtype = int) # indices, of length n = 6
sums = np.zeros(4) # target to sum up
sums[target] += values

我认为应该发生的事情非常清楚:value中的值应该在sums的{​​{1}}对应的索引处累加。

不幸的是,它并没有按预期工作,但并不奇怪。

在Numpy中实现它是否有效?

0 个答案:

没有答案