python - 总结 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}