gpt4 book ai didi

python - 为 numpy 数组赋值的花式索引

转载 作者:太空狗 更新时间:2023-10-30 02:10:39 25 4
gpt4 key购买 nike

我有一个二维 numpy 数组:

A = np.zeros(16).reshape(4,4)

我希望 (1, 1)、(1,3)、(3,1)(3,3) 单元格的值为1.

A[[1,3], [1:3]] = 1 

仅将 1 分配给 (1,1)(3,3)

A[[1,3], :][:, [1, 3]] = 1

不起作用,因为它制作的是数据副本而不是 View 。执行此操作的正确方法是什么?

最佳答案

使用 step=2 的切片:

A[1::2,1::2] = 1

或者显式传递所有索引:

A[[1,1,3,3],[1,3,1,3]] = 1

关于python - 为 numpy 数组赋值的花式索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/28663508/

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