Python 检查 numpy 数组中元组中的值是否失败

Question

我有一个返回元组的函数，我需要检查元组中是否存在特定元素。我不知道元组中的元素类型，但我知道我想要完全匹配。比如我想要

1 in (1, [0, 6], 0) --> True
1 in ([1], 0, 6]) --> False

这应该很简单，对吧？我只是检查

1 in tuple_output_from_function

。

如果元组中有一个 numpy 数组作为元素，则会中断

import numpy as np
s = tuple((2, [0, 6], 1))
4 in s --> False

t = tuple((2, np.arrary([0, 6]), 1))
4 in t --> ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我预计第二种情况会返回 False，因为 4 不在元组中。即使它在数组中，我仍然期望 False。我可以毫无错误地完成

0 in t[1]

。为什么会出现这种情况，以及如何在不假设存在 numpy 数组或必须显式检查它的情况下使我的检查稳健？

Answer 1

它会崩溃，因为如果你使用

in

运算符，Python 在底层正在使用相等运算符 (

==

)。

考虑一下：

t = (2, np.array([0, 6]), 1)
v = 1

print(v in t)

Python 检查元组

t

中的每个值是否相等。对于 numpy 数组，操作

2 == np.array([0, 6])

是另一个布尔数组

[False False]

。然后 Python 检查这个结果是否为真，这会抛出您看到的异常。

您可以使用

any()

并检查当前检查的值是否为该值的类型为

int

:

t = (2, np.array([0, 6]), 1)
v = 1

x = any(v == i for i in t if isinstance(i, int))
print(x)

打印：

True