python-从具有相同行的向量中给定索引替换numpy数组的列中的值

Question

A = np.zeros(shape = (3,4))
A = [[0 0 0 0]
     [0 0 0 0]
     [0 0 0 0]]

B = np.asarray[[2],[0],[3]]

如果没有 for 循环，是否有一种简单的方法来更改 A 中的 a 组件的值(例如更改为 1)，给定 B 中的索引
这样:

A = [[0 0 1 0]
     [1 0 0 0]
     [0 0 0 1]]

我已经能够通过 for 循环获得这个输出，但如果它不是为了高维数组的可伸缩性，我会更喜欢它。

Answer 1

看看 numpy 的 advanced indexing .
你可以做

A[[0, 1, 2], [2, 0, 3]] = 1

我们正在索引以获取 3 个位置并设置一个值，因此用于索引的列表每个都是 len 3。和 A有 2 个维度，因此我们使用两个列表(如果您想获取完整行，也可以只使用一个列表)。
它也适用于数组

A[np.arange(A.shape[0]), [2, 0, 3]] = 1