Элегантный способ проверить пустой результат numpy.nonzero

Question

Используя функцию numpy numpy.nonzero, есть ли элегантный способ проверить, являются ли кортежи как выходные пустые массивы?

В MATLAB, это очень легко

т.е.

answer = find(matrix_a < matrix_b); 
isempty(answer) 

Answer 1

numpythonic способ сделать это состоит в использовании any/all методы на ndarray объекты непосредственно.

В вашем примере, ваш код спрашивает: есть ли нет индексов, где matrix_a меньше, чем matrix_b?

not (matrix_a < matrix_b).any() 

Эквивалентно, имеет ли matrix_a все элементы, превышающие соответствующие элементы в matrix_b?

(matrix_a >= matrix_b).all() 

Answer 2

Для a (10,10) диапазона (100)

In [201]: np.nonzero(a>100) 
Out[201]: (array([], dtype=int32), array([], dtype=int32)) 

nonzero возвращает кортеж, с массивом для каждого измерения. Этот кортеж можно использовать для индексации всех элементов, где условие истинно.

Таким образом, вы можете проверить «пустой» nonzero, посмотрев на длину одного из этих массивов.

len(np.nonzero(a>98)[0])==0 

any на булевой маске появляется проще, хотя в быстрых тестах, это на самом деле медленнее.

np.any(a>98) 

MATLAB 'find' возвращает элементы, которые соответствуют. Обозначение numpy равно a[np.nonzero(a>100)] или с использованием булевой маски непосредственно a[a>100].

Таким образом, моя новая номинация на быстрый isempty(find...)) выглядит так: len(a[...])==0.