两个numpy ndarray的字典比较

Question

我无法找到一种简单的方法来比较两个（在我的情况下是多维）数组，以字典方式。

即

a = [1,2,3,4]
b = [4,0,1,6]

对于a < b，我希望得到true [true, false, false, true] 对于a > b，我希望得到false [false, true, true, false]

Answer 1

如果问题只是找到a是<还是>而不是b，那么以下内容应该有效。

def fn(a, b):
    # finds index of the first non matching element
    idx = np.where( (a>b) != (a<b) )[0][0]

    if a[idx] < b[idx]: print "a < b" 
    if a[idx] > b[idx]: print "a > b" 

Answer 2

与np.arange（4）[:: - 1] ** 2相乘，然后在该轴上求和。