NumPy库count_nonzero()函数
NumPy库的count_nonzero()函数是一个用于统计数组中非零元素个数的函数。它可以帮助我们快速计算出数组中非零元素的个数，从而更加快捷地完成非零元素的统计工作。
使用方法
使用NumPy库的count_nonzero()函数非常简单，只需要传入一个数组即可。下面是一个简单的使用示例：
import numpy as np a = np.array([[1, 0, 0], [0, 1, 0], [1, 0, 0]]) # 统计数组a中非零元素的个数 num_nonzero = np.count_nonzero(a) print(num_nonzero) # 输出结果为3
从上面的代码中可以看出，使用count_nonzero()函数只需要传入一个数组，就可以得到数组中非零元素的个数。
除了上面的这种使用方法外，count_nonzero()函数还支持传入多个参数，以实现更加复杂的统计工作。比如，我们可以传入一个数组以及一个axis参数，就可以实现对数组按指定轴进行统计：
import numpy as np a = np.array([[1, 0, 0], [0, 1, 0], [1, 0, 0]]) # 按行统计数组a中非零元素的个数 num_nonzero = np.count_nonzero(a, axis=1) print(num_nonzero) # 输出结果为[1 1 1]
从上面的代码可以看出，使用count_nonzero()函数时，可以传入一个axis参数，从而实现对数组按指定轴进行统计。
NumPy库的count_nonzero()函数可以帮助我们快速计算出数组中非零元素的个数，从而更加快捷地完成非零元素的统计工作。