Python Numpy count_nonzero函数

numpy.count_nonzero()函数计算数组arr中非零值的数量。

语法: numpy.count_nonzero(arr, axis=None)

参数 :
arr : [array_like] 用于计算非零的数组。
axis : [int or tuple, optional] 用于计算非零点的轴或轴的元组。默认为无，意味着非零点将沿着Arr的一个扁平化版本进行计数。

返回 : [int or array of int] 数组中沿给定轴的非零值的数量。否则，将返回数组中非零值的总数。

代码#1：

# Python program explaining
# numpy.count_nonzero() function
  
# importing numpy as geek 
import numpy as geek
  
arr = [[0, 1, 2, 3, 0], [0, 5, 6, 0, 7]]
  
gfg = geek.count_nonzero(arr)
  
print (gfg)

输出 :

代码#2：

# Python program explaining
# numpy.count_nonzero() function
  
# importing numpy as geek 
import numpy as geek
  
arr = [[0, 1, 2, 3, 4], [5, 0, 6, 0, 7]]
  
gfg = geek.count_nonzero(arr, axis = 0)
  
print (gfg)