用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)