Question

我想检查变量是否为None或numpy.array。我已经实现了check_a功能来执行此操作。

def check_a(a):
    if not a:
        print "please initialize a"

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

但是，这段代码引发了ValueError。什么是直接的方式？

ValueError                                Traceback (most recent call last)
<ipython-input-41-0201c81c185e> in <module>()
      6 check_a(a)
      7 a = np.array([1,2])
----> 8 check_a(a)

<ipython-input-41-0201c81c185e> in check_a(a)
      1 def check_a(a):
----> 2     if not a:
      3         print "please initialize a"
      4 
      5 a = None

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

Answer 1

使用not a来测试a是None是否假设a的其他可能值的真值为True。但是，大多数NumPy数组根本没有真值，not无法应用于它们。

如果您想测试对象是否为None，最通用，最可靠的方法是对is进行None检查：

if a is None:
    ...
else:
    ...

这并不依赖于具有真值的对象，因此它适用于NumPy数组。

请注意，测试必须是is，而不是==。 is是对象标识测试。 ==是参数所说的，NumPy数组表示它是一个广播的元素相等比较，产生一个布尔数组：

>>> a = numpy.arange(5)
>>> a == None
array([False, False, False, False, False])
>>> if a == None:
...     pass
...
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: The truth value of an array with more than one element is ambiguous.
 Use a.any() or a.all()

另一方面，如果你想测试一个对象是否是一个NumPy数组，你可以测试它的类型：

# Careful - the type is np.ndarray, not np.array. np.array is a factory function.
if type(a) is np.ndarray:
    ...
else:
    ...

您还可以使用isinstance，它还会为该类型的子类返回True（如果这是您想要的）。考虑到np.matrix是多么可怕和不兼容，你可能实际上并不想要这个：

# Again, ndarray, not array, because array is a factory function.
if isinstance(a, np.ndarray):
    ...
else:
    ...

Answer 2

您可以查看对象是否具有形状

def check_array(x):
    try:
        x.shape
        return True
    except:
        return False

Answer 3

如果您尝试做类似a is not None的事情，则会出现相同的问题：Numpy抱怨一个人必须使用a.any或a.all。一种解决方法是：

if not (a is None):
    pass

不太漂亮，但是确实可以。

Answer 4

坚持==而不考虑其他类型，以下也是可能的。
type(a) == type(None)

检查变量是否为None或numpy.array

4 个答案: