我有两个数组A
和B
:
A=array([[ 5., 5., 5.],
[ 8., 9., 9.]])
B=array([[ 1., 1., 2.],
[ 3., 2., 1.]])
任何有“1”的地方B
我都想对A
.
因此,例如对于这个答案将是 5+5+9=10
我希望这继续 2,3....n (所有唯一值B
)
所以对于 2 的......它将是 9+5=14 而对于 3 的它将是 8
我使用以下方法找到了唯一值:
numpy.unique(B)
我意识到这需要多个步骤,但我无法真正使用索引矩阵将这些位置汇总到另一个矩阵中。