如何解决替换从相同行的向量给定索引的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
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。