gpt4 book ai didi

python - Numpy:根据列索引数组设置每行的 1 个元素

转载 作者:行者123 更新时间:2023-12-04 01:53:45 25 4
gpt4 key购买 nike

<分区>

假设我有以下数组:

a = np.zeros([4,3])
b = np.asarray([0,1,2,1])

现在如何根据数组 b 指示的列索引将数组 a 中每一行的元素设置为 1?我需要的是一个如下所示的数组 c:

[[ 1.  0.  0.]
[ 0. 1. 0.]
[ 0. 0. 1.]
[ 0. 1. 0.]]

或者,有没有办法直接将数组 b 转换为数组 c?

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com