Python numpy.count_nonzero函数方法的使用

numpy.count_nonzero

numpy.count_nonzero(a, axis=None, *, keepdims=False)     [source]

计算数组a中非零值的数量。

“non-zero”一词是指Python 2.x内置方法__nonzero__()（重命名为__bool__()。 例如，如果任何数字非零，则将其视为真实，而如果其不是空字符串，则将其视为真实。 因此，此函数（递归）计算a（及其子数组）中有多少个元素具__nonzero__()或__bool__()方法计算为True。

例子

>>> np.count_nonzero(np.eye(4))
4
>>> a = np.array([[0, 1, 7, 0],
...               [3, 0, 2, 19]])
>>> np.count_nonzero(a)
5
>>> np.count_nonzero(a, axis=0)
array([1, 1, 2, 1])
>>> np.count_nonzero(a, axis=1)
array([2, 3])
>>> np.count_nonzero(a, axis=1, keepdims=True)
array([[2],
       [3]])