Python numpy count_nonzero用法及代码示例

本文简要介绍 python 语言中 numpy.count_nonzero 的用法。

用法:

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

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

“non-zero” 这个词是指 Python 2.x 内置方法 __nonzero__()(在 Python 3.x 中重命名为 __bool__())的 Python 对象，用于测试对象的 “truthfulness”。例如，任何数字如果非零则被认为是真实的，而任何字符串如果不是空字符串则被认为是真实的。因此，此函数(递归)计算 a(及其子数组中)有多少元素的 __nonzero__() 或 __bool__() 方法被评估为 True 。

参数：

a： array_like

要计算非零的数组。

axis： int 或元组，可选

沿其计数非零的轴或轴元组。默认为无，这意味着非零将沿 a 的扁平版本计算。

keepdims： 布尔型，可选

如果将其设置为 True，则计算的轴将作为尺寸为 1 的尺寸留在结果中。使用此选项，结果将针对输入数组正确广播。

返回：

count： int 或 int 数组

数组中沿给定轴的非零值的数量。否则，返回数组中非零值的总数。

例子：

>>> 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]])