与重载运算符的numpy元素比较

Question

我有两个numpy数组包含带有重载比较运算符的对象，它返回另一个对象，而不是True或False。如何创建包含单个比较结果的数组。我想结果是一个对象数组，如下面的

lhs = ... # np.array of objects with __le__ overloaded
rhs = ... # another np.array
result = np.array([l <= r for l, r in izip(lhs, rhs)])

但lhs <= rhs给了我一系列布尔。 有没有办法让result成为__le__方法调用结果的数组，而无需编写python循环？

Answer 1

ndarray上的{p> Numpy's Github page表示比较运算符等同于Numpy中的 ufunc 表单。因此，lhs <= rhs相当于np.less_equal(lhs, rhs)

从np.info(np.less_equal)

的输出

 Returns
 ------- out : bool or ndarray of bool
   Array of bools, or a single bool if `x1` and `x2` are scalars.

要解决此问题，您可以使用：

import operator
result = np.vectorize(operator.le)(lhs, rhs)

np.vectorize允许您在比较中使用Numpy的广播。它将使用您的对象比较，并将以与列表推导相同的形式返回这些比较结果的numpy数组。