Numpy:如何计算等于 true 的元素数量


您可以使用以下基本语法来计算 NumPy 数组中等于True的元素数量:

 import numpy as np

n.p. count_nonzero (my_array)

此特定示例将返回名为my_array的 NumPy 数组中等于True的元素数量。

以下示例展示了如何在实践中使用此语法。

示例:计算 NumPy 数组中等于 True 的元素数量

以下代码演示如何使用count_nonzero()函数来计算 NumPy 数组中等于 True 的元素数量:

 import numpy as np

#create NumPy array
my_array = np. array ([True, False, False, False, True, True, False, True, True])

#count number of values in array equal to True
n.p. count_nonzero (my_array)

5

从输出中,我们可以看到 NumPy 数组的5 个值等于True

我们可以手动查看 NumPy 数组来验证数组中确实有三个元素为True

如果您想计算等于False 的元素数量,可以从size()函数中减去count_nonzero()函数的结果,如下所示:

 import numpy as np

#create NumPy array
my_array = np. array ([True, False, False, False, True, True, False, True, True])

#count number of values in array equal to False
n.p. size (my_array) - np. count_nonzero (my_array)

4

从输出中,我们可以看到 NumPy 数组的4 个值等于False

注意:如果 NumPy 数组中有 NaN 值,则 count_nonzero()函数会将每个 NaN 值视为等于 True 的元素。

其他资源

以下教程解释了如何在 Python 中执行其他常见操作:

如何计算 NumPy 数组的众数
如何将函数映射到 NumPy 数组
如何按列对 NumPy 数组进行排序

添加评论

您的电子邮箱地址不会被公开。 必填项已用*标注