Подсчет значений в функции с Python - PullRequest
0 голосов
/ 02 ноября 2018

У меня есть следующая функция

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. Эти значения сначала агрегируются. Теперь я хочу узнать, сколько этих значений агрегировано.

1 Ответ

0 голосов
/ 02 ноября 2018

Я нашел решение:

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)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...