如何仅将值添加到 Numpy 2D 矩阵中的某些索引答案

【问题标题】：How to add values to only certain index in Numpy 2D Matrix如何仅将值添加到 Numpy 2D 矩阵中的某些索引
【发布时间】：2022-01-14 03:31:00
【问题描述】：

我有一个二维矩阵

matrix = np.array([[1,2],[3,4],[5,6]])
index = np.array([0, 1, 1])
add_value = np.array([1, 2, 3])

我想将 add_value 添加到矩阵，但只添加到索引列表中与索引对应的元素。例如add_value中的1应该加到[1,2]中的第一个元素，即1，结果为2。所以输出应该是

np.array([[2,2],[3,6],[5,9]])

【问题讨论】：

【解决方案1】：

使用简单的多维索引：

matrix[np.arange(matrix.shape[0]), index] += add_value

或者使用 python 内建函数：

matrix[tuple(zip(*enumerate(index)))] += add_value

输出：

array([[2, 2],
       [3, 6],
       [5, 9]])

【讨论】：

【解决方案2】：

for i, x in enumerate(add_value):
    matrix[i][index[i]] += x

【讨论】：