文章/答案/技术大牛

发布

社区首页 >问答首页 >检查二维numpy数组中的每个元素是否都在集合中

问检查二维numpy数组中的每个元素是否都在集合中
EN

Stack Overflow用户

提问于 2021-11-01 00:21:14

回答 1查看 40关注 0票数 0

我希望对numpy数组中的每个元素有效地检查该元素是否存在于一个集合中。例如：

segmask = np.array([[1,2,3,4]])
numbers = {2,4}
check_if_in_set(numbers, segmask)

应该返回

[[False  True False  True]]

或者类似的。

发布于 2021-11-01 00:21:14

np.isin(segmask, list(numbers))

会给你你想要的结果。根据the isin documentation的说法，在将集合提供给isin函数之前，必须将其转换为列表。

票数 2

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/69791575

复制

相似问题

问检查二维numpy数组中的每个元素是否都在集合中EN