Как проверить, больше ли число единиц, чем число нулей в массиве numpy, не обходя весь массив? - PullRequest
0 голосов
/ 12 февраля 2020

Учитывая логический numpy nd-массив, как я могу найти, если общее число ones больше, чем общее число zeros в массиве, не обходя весь массив с вложенными циклами for. Я имел в виду функцию, встроенную в any() и all(). Скажем max_bool(), который работает следующим образом:

def max_bool(array):
    return array.ones => array.zeros

Обход не является вариантом, так как размеры массивов, с которыми я собирался работать, имеют разные непредсказуемые размеры и могут быть слишком большими. Меня не беспокоит точное число ones & zeros. Просто если массив имеет больше ones или zeros, даже если число ones на единицу больше, чем число zeros. Любая помощь?

1 Ответ

3 голосов
/ 12 февраля 2020

Самый простой способ, которым я могу придумать:

def max_bool(array):
    return array.mean() >= .5
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...