检查变量是None还是numpy.array

Question

我在表中查找键是否有关联数组。通过设计，mytable.__getitem__()somtimes返回None，而不是KeyError-s。我希望此值是None，或者是与w关联的numpy数组。

value = table[w] or table[w.lower()]
# value should be a numpy array, or None
if value is not None:
    stack = np.vstack((stack, value))

只有当我使用上面的代码，并且第一次查找是匹配的，我才能得到：

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

如果我使用value = table[w].any() or table[w.lower()].any()，那么如果它不匹配，我会遇到：

AttributeError: 'NoneType' object has no attribute 'any'

我一定是错过了正确的方法，怎么办？

Answer 1

这应该管用：

value = table[w]
if value is None:
    value = table[w.lower()]