Python Numpy：如何计算bool数组中真正元素的数量 [英] Python Numpy: how to count the number of true elements in a bool array

问题描述

我有一个NumPy数组boolarr的布尔类型。我想计算其值为 True 的元素的数量。是否有专门用于此任务的NumPy或Python例程？或者，我需要迭代我的脚本中的元素吗？

I have a NumPy array 'boolarr' of boolean type. I want to count the number of elements whose values are True. Is there a NumPy or Python routine dedicated for this task? Or, do I need to iterate over the elements in my script?

推荐答案

两个选项如下。

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)
>>> boolarr
array([[False, False,  True],
       [ True, False,  True],
       [ True, False,  True]], dtype=bool)

>>> np.sum(boolarr)
5

当然， c> bool - 具体答案。更一般地，您可以使用 numpy.count_nonzero 。

Of course, that is a bool-specific answer. More generally, you can use numpy.count_nonzero.

>>> np.count_nonzero(boolarr)
5

这篇关于Python Numpy：如何计算bool数组中真正元素的数量的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！