numpy代码统计数组非零值个数

可以使用 numpy 中的 count_nonzero() 函数来统计数组中非零值的个数。

例如：

import numpy as np

a = np.array([[1, 0, 0], [0, 2, 0], [1, 1, 0]])
count = np.count_nonzero(a)
print(count)  # 输出：4

上面的代码会将二维数组 a 中的所有非零值(即 1 和 2)的个数统计出来，结果为 4。

如果希望在某一维上进行统计，可以指定 axis 参数，例如：

count = np.count_nonzero(a, axis=0)
print(count)  # 输出：[2 1 0]

count = np.count_nonzero(a, axis=1)
print(count)  # 输出：[1 1 1]

上面的代码会在每一列上(axis=0)或每一行上(axis=1)统计非零值的个数。