Сравнивая 2 массива используя NumPy и возвращая индекс с меньшим числом

Question

У меня есть два массива:

A = array([1.48067671, 1.48352177, 1.4481969 ,1.00006692, 1.28575167,1.32976551])
B = array([1.364952  , 1.36871978, 1.33351794,1.17396389, 1.19171093,1.23191085])

Они имеют одинаковую длину - мне нужно сравнить каждый индекс и определить, какой массив имеет меньшее значение. Например - приведенный выше результат должен быть:

[B,B,B,A,B,B]

Answer 1

Вы можете использовать логическую маску и привести к int:

(B < A).astype(int)