Функция count_nonzero
Функция count_nonzero подсчитывает количество ненулевых
элементов в массиве NumPy. Первым параметром функция принимает
массив. Вторым параметром можно указать ось, вдоль которой
будет производиться подсчет. Функция возвращает целое число
или массив чисел, в зависимости от переданных параметров.
Под ненулевыми элементами понимаются значения, которые не равны
нулю, None или False.
Синтаксис
np.count_nonzero(arr, [axis])
Пример
Давайте подсчитаем количество ненулевых элементов в одномерном массиве:
import numpy as np
arr = np.array([0, 1, 2, 0, 3, 4, 0, 5])
res = np.count_nonzero(arr)
print(res)
Результат выполнения кода:
5
Пример
Давайте подсчитаем ненулевые элементы в двумерном массиве:
import numpy as np
arr = np.array([[0, 1, 2], [3, 0, 4], [5, 6, 0]])
res = np.count_nonzero(arr)
print(res)
Результат выполнения кода:
6
Пример
Давайте подсчитаем ненулевые элементы вдоль указанной оси:
import numpy as np
arr = np.array([[0, 1, 2], [3, 0, 4], [5, 6, 0]])
res = np.count_nonzero(arr, axis=0)
print(res)
Результат выполнения кода:
[2 2 2]
Пример
Давайте подсчитаем ненулевые элементы вдоль второй оси:
import numpy as np
arr = np.array([[0, 1, 2], [3, 0, 4], [5, 6, 0]])
res = np.count_nonzero(arr, axis=1)
print(res)
Результат выполнения кода:
[2 2 2]
Пример
Давайте подсчитаем ненулевые элементы в трёхмерном массиве:
import numpy as np
arr = np.array([[[0, 1], [2, 0]], [[3, 4], [0, 5]]])
res = np.count_nonzero(arr)
print(res)
Результат выполнения кода:
5
Пример
Давайте подсчитаем ненулевые элементы в массиве с булевыми значениями:
import numpy as np
arr = np.array([True, False, True, False, True])
res = np.count_nonzero(arr)
print(res)
Результат выполнения кода:
3
