gpt4 book ai didi

python - 获取满足某些条件的 numpy 数组的元素

转载 作者:太空宇宙 更新时间:2023-11-03 10:49:58 25 4
gpt4 key购买 nike

<分区>

假设我有一个充满整数的 numpy 数组

[[1,2],[3,4]]

如果元素满足属于列表,我想得到一个包含 1 的二进制数组,否则为 0。

如果我写

condition = arr == 2

我明白了

[[false, true], [false, false]]

这就是我想要的。

但是如果我想保留元素 2 和 3 怎么办?我试过了

condition = arr in [2,3]

但它不起作用,我得到一个

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

我想对任何可能的列表执行此操作。

有什么有效的方法吗?我知道如何残酷地做到这一点,但我需要高效。

非常感谢!

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