numpy.count_nonzero#

numpy.count_nonzero(a, axis=None, *, keepdims=False)[源代码]#

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

术语“非零”是指 Python 对象中用于测试对象“真值”的 Python 2.x 内置方法 __nonzero__()（在 Python 3.x 中重命名为 __bool__()）。例如，任何非零数字都被认为是真值，而任何非空字符串也被认为是真值。因此，此函数（递归地）计算 a 中（以及其中的子数组中）有多少个元素的 __nonzero__() 或 __bool__() 方法被评估为 True。

参数:

a类数组: 要计算非零值的数组。
axis整型或元组，可选: 沿其计算非零值的轴或轴元组。默认值为 None，表示将沿 a 的展平版本计算非零值。
keepdims布尔值，可选: 如果设置为 True，则被计数的轴将作为大小为一的维度保留在结果中。使用此选项，结果将与输入数组正确地进行广播。

返回:

count整型或整型数组: 数组沿给定轴的非零值数量。否则，返回数组中非零值的总数量。

另请参阅

nonzero: 返回所有非零值的坐标。

示例

>>> import numpy as np
>>> 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]])