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給我的bool數組。 有沒有辦法去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的廣播在比較過。它將使用您的對象比較，並將以與列表理解相同的形式返回這些比較結果的數組。