当需要统计数组中非零元素的个数时,可以使用 Python Numpy 库中的 count_nonzero
函数。该函数用于返回数组中非零元素的数量。
函数定义如下:
numpy.count_nonzero(a, axis=None)
其中,参数 a
表示输入的数组,参数 axis
表示统计非零元素的轴,如果设置为 None
,则统计整个数组中的非零元素。
下面是两个示例说明:
示例一
import numpy as np
a = np.array([[1, 0], [2, 3]])
print(np.count_nonzero(a))
输出结果是 3
。因为数组 a
中非零元素的个数为 3
(即 1
,2
,3
)。
示例二
import numpy as np
a = np.array([[1, 0], [2, 3]])
print(np.count_nonzero(a, axis=0))
输出结果是 [2 1]
。因为数组 a
沿着轴 0
统计非零元素,即每一列中非零元素的个数。
以上就是 Python Numpy count_nonzero
函数的完整攻略。需要注意的是,该函数只能处理 Numpy 数组,对于普通的 Python 列表需要先转换为 Numpy 数组才能使用。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Python Numpy count_nonzero函数 - Python技术站