Numpy: как посчитать количество элементов, равных true

Вы можете использовать следующий базовый синтаксис для подсчета количества элементов, равных True , в массиве NumPy:

 import numpy as np

n.p. count_nonzero (my_array)

Этот конкретный пример вернет количество элементов, равное True, в массиве NumPy с именем my_array .

В следующем примере показано, как использовать этот синтаксис на практике.

Пример: подсчитать количество элементов, равных True, в массиве NumPy.

Следующий код показывает, как использовать функцию count_nonzero() для подсчета количества элементов в массиве NumPy, равного True:

 import numpy as np

#create NumPy array
my_array = np. array ([True, False, False, False, True, True, False, True, True])

#count number of values in array equal to True
n.p. count_nonzero (my_array)

5

Из вывода мы видим, что 5 значений массива NumPy равны True .

Мы можем вручную просмотреть массив NumPy, чтобы убедиться, что в массиве действительно есть три элемента со значением True .

Если вместо этого вы хотите подсчитать количество элементов, равное False , вы можете вычесть результаты функции count_nonzero() из функции size() следующим образом:

 import numpy as np

#create NumPy array
my_array = np. array ([True, False, False, False, True, True, False, True, True])

#count number of values in array equal to False
n.p. size (my_array) - np. count_nonzero (my_array)

4

Из вывода мы видим, что 4 значения массива NumPy равны False .

Примечание . Если в вашем массиве NumPy есть значения NaN, функция count_nonzero() будет считать каждое значение NaN как элемент, равный True.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные операции в Python:

Как рассчитать режим массива NumPy
Как сопоставить функцию с массивом NumPy
Как отсортировать массив NumPy по столбцу