gpt4 book ai didi

python - Numpy 数组,指定返回哪些元素

转载 作者:行者123 更新时间:2023-11-30 23:08:09 25 4
gpt4 key购买 nike

假设我有以下 5x5 numpy 数组,名为 A

array([[6, 7, 7, 7, 8],
[4, 2, 5, 5, 9],
[1, 2, 4, 7, 4],
[0, 7, 3, 6, 8],
[4, 9, 6, 1, 6]])

这个 5x5 数组称为 F

array([[1,0,0,0,0],
[1,0,0,0,0],
[1,0,0,0,0],
[1,0,0,0,0],
[0,0,0,0,0]])

我一直在尝试使用 np.copyto,但我无法理解为什么它不起作用/它是如何工作的。ValueError:无法从形状(5,5)广播输入数组变成形状(2)

是否有一种简单的方法可以仅获取 F 中具有相应 1 的匹配整数的值(当它们覆盖在 A 上时)?即它将返回,6,4,1,0

最佳答案

你可以做这个小技巧:A[F==1]

In [8]:

A[F==1]
Out[8]:
array([6, 4, 1, 0])

查看Boolean indexing

关于python - Numpy 数组,指定返回哪些元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31886674/

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