检查单个元素是否包含在 Numpy 数组中

Question

我只想检查一个 numpy 数组是否快速包含一个数字，类似于列表的包含。有没有一种简洁的方法可以做到这一点？

 a = np.array(9,2,7,0)
a.contains(0)  == true

原文由 pd109 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

您可以使用 0 in a 。 IE

 a = np.array([9,2,7,0])
0 in a

原文由 Bharath M Shetty 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

如果 a 是一个 numpy 数组：

 a = np.array([1, 2])

然后使用：

 1 in a

返回 true，同时：

 0 in a

返回假

原文由 Johnnyh101 发布，翻译遵循 CC BY-SA 3.0 许可协议