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 за стовпцями