按 python 中的索引将矩阵 a 中的行添加到矩阵 b

Question

我正在尝试以更有效（矢量化）的方式而不是循环来进行以下计算。

for i in range(N):

  y[:,idx[i]] -= X[i]

在哪里：

• X是 N 乘 D 矩阵，
• y是 k 乘 D 矩阵
• idx是长度为 N 的向量，其值介于 0 和 k-1 之间

例子：

x = array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15],
       [16, 17, 18, 19]])


y =  array([[100, 101, 102, 103],
           [104, 105, 106, 107],
           [108, 109, 110, 111]])

idx = array([2, 1, 1, 1, 2])

所以 output 应该是：

y = array([[100, 101, 102, 103],
          [ 80,  78,  76,  74],
          [ 92,  91,  90,  89]])

这相当于：

>>> for i in range(5):
...     y[idx[i],:] -= x[i]

Answer 1

Pandas 解决方案最短到 go。 您可以通过 NumPy 来完成，但代码要长一点：

import pandas as pd
df = pd.DataFrame(x, idx)
s = df.groupby(df.index).sum()
y[s.index] -= s

或者在您的情况下，由于索引是索引，您可以直接使用：

df = pd.DataFrame(x)
s = df.groupby(idx).sum()
y[s.index] -= s

output：

array([[100, 101, 102, 103],
       [ 80,  78,  76,  74],
       [ 92,  91,  90,  89]])