用python计算函数中的值

Question

我有以下功能

def sum_NE(data, i, col='VALUES'):
   return data.iloc[get_NE(i, len(data))][col].sum()

这很好。 但我想再做一件事。 VALUES列包含零和大于零的值。 如何计算评估sum()时使用的所有大于零的值？ 函数get_NE返回一个列表。 我尝试了下面的代码，但是没有用。

def sum_NE(data, i, col='VALUES'):
    return data.iloc[get_NE(i, len(data))][col].count()

函数get_NE是一个返回列表的函数。 例如[5、6、8、12]。 这些值是data帧中的行，使用[col]参考，我正在VALUES列中查看某些值。 这些值首先进行汇总。 现在，我想找出这些值中有多少个被汇总。

Answer 1

我找到了解决方案：

def sum_NE(data, i, col='VALUES'):
    return sum(1 for i in data.iloc[get_NE(i, len(data))][col] if float(i) > 0)