gpt4 book ai didi

python - python 中用于 nd 数组的逻辑索引

转载 作者:太空宇宙 更新时间:2023-11-04 10:13:31 28 4
gpt4 key购买 nike

我正在尝试从 (N x N x N) numpy 数组中提取所有索引,其中 A 和 B 数组中的值都等于某个值 x - 找到共同的重叠部分。

我正在尝试:

   A[A==1 and B==1]

但是报错:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我该如何解决这个问题?

最佳答案

Numpy 不能重载“and”关键字。但是,它为此重载了二进制 AND 运算符 &。尝试:

A[(A==1) & (B==1)]

括号很重要。我发现它经常(不总是)比 logical_and 更易读

关于python - python 中用于 nd 数组的逻辑索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36839515/

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