博客

Python Numpy是一款用于科学计算的第三方库，其中count_nonzero函数用于统计数组中非零元素的个数。

函数语法

numpy.count_nonzero(arr, axis=None)

参数说明

• arr：需要统计非零元素的数组
• axis：可选参数，表示沿着哪个轴统计非零元素。默认为None，表示统计所有的元素。

返回值

返回一个整数，表示数组中非零元素的个数。

示例说明

示例1：统计一维数组中非零元素的个数

import numpy as np

arr = np.array([0, 1, 2, 0, 3, 0])

count = np.count_nonzero(arr)

print(count) # 输出3

示例2：统计二维数组中每行非零元素的个数

import numpy as np

arr = np.array([[0, 1, 2], [0, 0, 0], [0, 3, 0]])

count = np.count_nonzero(arr, axis=1)

print(count) # 输出[2, 0, 1]

上述示例中，示例1统计了一维数组arr中的非零元素个数，结果为3。示例2统计了二维数组arr中每一行的非零元素个数，结果为[2, 0, 1]。其中，axis参数被设置为1，表示沿着每一行统计非零元素的个数。