Лексикографическое сравнение двух numpy ndarrays

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

Multiply с np.arange (4) [:: - 1] * * 2 и затем суммировать по этой оси.