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

我有两个包含带有重载比较运算符的对象的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给了我一系列的bools.

有没有办法在没有编写python循环的情况下将结果作为__le__方法调用结果的数组？

解决方法:

ndarray上的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数组.

标签：python,operator-overloading,numpy