重复 numpy 操作的预期行为

Question

我在寻找关于 numpy 数组重复操作的问题时找到了这个答案：Increment Numpy multi-d array with repeated indices。我现在的问题是，为什么会出现这种行为。

import numpy as np
t = np.eye(4)

t[[0,0,1],[0,0,1]]

导致

array([1.,1.,1.])

不应该

t[[0,0,1],[0,0,1]]+=1 

导致

[[3,0,0,0],
 [0,2,0,0],
 [0,0,1,0],
 [0,0,0,1]]

?

Answer 1

请参阅documentation 了解数组索引以及基本索引和高级索引之间的区别。

t[[0,0,1],[0,0,1]] 属于高级索引类别，如文档中所述：

高级索引始终返回数据的副本（与返回视图的基本切片相反）。

副本在第一次增量之前被评估，所以正如预期的那样，

import numpy as np
t = np.eye(4)
t[[0,0,1],[0,0,1]] += 1
print(t)

打印：

[[ 2.  0.  0.  0.]
 [ 0.  2.  0.  0.]
 [ 0.  0.  1.  0.]
 [ 0.  0.  0.  1.]]

根据上面的 cmets，使用 numpy.ufunc.at 或 numpy.add.at 来解决这个问题。