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

问题描述

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循环获得此输出，但是如果不能在更高维数组中实现可伸缩性，则更希望使用它。

解决方法

看看numpy的advanced indexing。

你可以做

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

我们正在索引以获取3个位置并设置一个值，因此用于索引的列表各为len 3。A有2个暗角，因此我们使用两个列表（如果满足以下条件，则也只能使用一个列表您想获取完整的行）。

它也适用于数组

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