python - NumPy，为什么相等检查不适用于对象数组？

Question

使用以下对象数组：

a = np.array([[1], [1, 2], [1, 2, 3], [1], [1]], dtype=object)
b = np.array([(1,), (1, 2), (1, 2, 3), (1,), (1,)], dtype=object)

以下相等性检查不起作用：

a==[1]
#array([False, False, False, False, False], dtype=bool)

b==(1,)
#array([False, False, False, False, False], dtype=bool)

如果我改用字符串：

c = np.array(['[1]', '[1, 2]', '[1, 2, 3]', '[1]', '[1]'])

平等检查有效：

c == '[1]'
#array([ True, False, False,  True,  True], dtype=bool)

为什么数组检查的行为是这样的？

如果我们遍历 a 或 b 并执行检查，它也会给出预期的结果：

[i==[1] for i in a]
#[True, False, False, True, True]

[i==(1,) for i in b]
#[True, False, False, True, True]

谢谢！

score 6 · Accepted Answer

NumPy 旨在在许多情况下自动将类似数组的对象视为数组。在这里，NumPy 看到它[1]并且(1,)是类似数组的对象并应用广播规则。任一侧的 Length-1 轴扩展为另一个对象对应轴的长度，如果一个对象的尺寸小于另一个，则在左侧用其他对象在这些尺寸中的长度填充缺失的尺寸。因此，

a == [1]

给出相同的结果

a == numpy.array([1, 1, 1, 1, 1])

这是一个 5 Falses 的数组。

1 回答 1