汇总Pandas数据帧中的列的部分

Question

假设我的Pandas数据帧中有两列

Category   Price
Weekend       30
Weekend       20
Thursday      10
Wednesday      5
Weekend       55

我想要总结类别并添加值。例如，我想将"Weekend"的所有值相加，并将其存储在字典中。所以我有一些类似的东西：

summary = {"Weekend":105, "Thursday": 10, "Wednesday": 5}

我该如何轻松地做到这一点呢？

Answer 1

您可以先使用.groupby()，然后使用.sum()。

>>> df = pd.DataFrame(dict(Category=['Weekend', 'Weekend', 
                                     'Thursday', 'Wednesday', 'Weekend'], 
                           Price=[30, 20, 10, 5, 55]))

>>> df.groupby('Category').sum()
           Price
Category        
Thursday      10
Wednesday      5
Weekend      105

>>> df.groupby('Category').sum().Price.to_dict()
{'Thursday': 10, 'Wednesday': 5, 'Weekend': 105}