在numpy数组中多次递增值

时间:2017-04-14 23:10:01

标签: python python-3.x numpy

我试图多次增加numpy数组中的值。例如,我想这样做:

input = np.array([[1,1],[2,2]])
input[[0,0,0],[1,0,1]] += [4,1,6]

并获得输出:

array([[2, 11], [2, 2]])

然而,numpy给出了

array([[2, 7], [2, 2]])

有没有办法多次递增相同的索引?

0 个答案:

没有答案