汇总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}