numpy 类似countif的函数

在NumPy中，可以使用函数`numpy.count_nonzero()`来实现类似于Excel中的`COUNTIF`函数的功能，该函数可以统计数组中满足条件的元素个数。

例如，假设有一个一维数组`arr`，我们想要统计其中大于等于10的元素个数，代码如下：

import numpy as np

arr = np.array([2, 6, 10, 12, 15, 18, 22])
count = np.count_nonzero(arr >= 10)
print(count)  # 输出：5

在上面的代码中，`arr >= 10`会返回一个布尔类型的数组，其中满足条件的元素为`True`，不满足条件的元素为`False`。然后，`np.count_nonzero()`函数会统计数组中`True`的个数，即大于等于10的元素个数。

除了`count_nonzero()`函数，还可以使用`np.sum()`函数实现类似的功能，例如：

count = np.sum(arr >= 10)
print(count)  # 输出：5

这两个函数的区别在于，`count_nonzero()`函数只统计数组中满足条件的元素个数，而`sum()`函数会将满足条件的元素的值相加。