Python广播：在一维二维数组中使用一维向量作为索引

Question

假设我有一个形状table的{​​{1}}二维数组。

假设我有1D数组，m x n形状col_indices，范围m就值而言。

最后，假设我有一个[0,n)形状vals的值数组，由u索引。

在伪代码中，我想写：

i

并且，手动写出：

table[:,col_ind[:]] += vals[i]

Answer 1

使用整数数组替换行切片来使用advanced indexing：

t[np.arange(c.size), c] = v[i]

t
#array([[10,  0,  0],
#       [ 0, 10,  0],
#       [ 0,  0, 10]])