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 中执行其他常见操作: