如何计算NumPy bool数组中真实元素的数量

如何计算NumPy bool数组中真实元素的数量

您有多种选择。以下是两个选项。

numpy.sum(boolarr)numpy.count_nonzero(boolarr)

这是一个例子：

>>> import numpy as np>>> boolarr = np.array([[0, 0, 1], [1, 0, 1], [1, 0, 1]], dtype=np.bool)>>> boolarrarray([[False, False,  True],       [ True, False,  True],       [ True, False,  True]], dtype=bool)>>> np.sum(boolarr)5

当然，这是一个

bool

特定的答案。通常，您可以使用

numpy.count_nonzero

。

>>> np.count_nonzero(boolarr)5