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

Question

我有两个包含带有重载比较运算符的对象的 numpy 数组，returns 另一个对象，而不是 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

Numpy's Github page on ndarray 声明比较运算符等同于 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 的广播。它将使用您的对象比较，并将 return 这些比较结果的 numpy 数组，其形式与您的列表理解相同。

numpy elementwise comparison with overloaded operator