Pandon Pandas подсчет и суммирование конкретных условий

Question

Существуют ли в пандах отдельные функции для выполнения эквивалентов SUMIF, которые суммируются по определенному условию и COUNTIF, который подсчитывает значения конкретных условий из Excel?

Я знаю, что есть много несколько ступенчатых функций, которые могут быть использованы для

, например, для sumif я могу использовать (df.map(lambda x: condition), or df.size()) затем использовать .sum()

и для countif я могу использовать (groupby functions и искать мой ответ или используйте фильтр и .count())

Есть ли простой одноэтапный процесс для выполнения этих функций, когда вы вводите условие и фрейм данных, и получаете суммы или подсчитываете результаты?

Answer 1

Вы можете сначала сделать условный выбор и подвести итоги выбора с помощью функции sum.

>> df = pd.DataFrame({'a': [1, 2, 3]}) 
>> df[df.a > 1].sum() 
a 5 
dtype: int64 

Имея более чем одно условие:

>> df[(df.a > 1) & (df.a < 3)].sum() 
a 2 
dtype: int64 

Answer 2

Вы не говоря уже о возможности фантазии индексации dataframes, например .:

>>> df = pd.DataFrame({"class":[1,1,1,2,2], "value":[1,2,3,4,5]}) 
>>> df[df["class"]==1].sum() 
class 3 
value 6 
dtype: int64 
>>> df[df["class"]==1].sum()["value"] 
6 
>>> df[df["class"]==1].count()["value"] 
3 

Вы могли бы заменить df["class"]==1 другим условием.