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）统计非零值的个数。