python - 想要计算列中满足条件的值的数量

Question

我正在尝试计算列中满足特定条件(例如，大于 0.75)的值的数量。我的列由 2000 多个小数组成。

这是我试过的，

a = len(fs)
c = np.zeros(a)

for i in fs[0:a]:
    if i >= 0.75:
        print = 1

    elif i < 0.75:
        print = 0 

fs 是我的专栏。

此代码正确打印了我想要的 0 和 1，但我不确定如何计算打印的 1 的数量。我想首先制作一个零数组，然后以某种方式将数组附加到循环中以获得正确的 0 和 1 的数组。然后我可以对数组求和。我不太确定如何解决这个问题，我尝试的一切都不起作用(我在编程方面非常缺乏经验)。有没有人对如何去做有任何建议？我知道这很简单...

谢谢!

Answer 1

在 numpy 中你可以这样做:

np.where(fs >= 0.75)[0].size

或

np.count_nonzero(fs >= 0.75)

两者是等价的，但我可能更喜欢第二种。请参阅文档以获取解释:

http://docs.scipy.org/doc/numpy/reference/generated/numpy.count_nonzero.html

但基本上 fs >= 0.75 创建一个与 fs 长度相同的 bool 数组，其中它的元素是 True 或 False 基于条件。由于这分别相当于 1 和 0，因此 np.count_nonzero 会返回非零元素的计数。

当然，您也可以对 fs 进行切片:

np.count_nonzero(fs[0:a] >= 0.75)