假设我的Pandas数据框中有两列
Category Price
Weekend 30
Weekend 20
Thursday 10
Wednesday 5
Weekend 55
我想总结类别并添加值。我想将
"Weekend"
的所有值加在一起,并将其存储在例如字典中。所以我有这样的事情:
summary = {"Weekend":105, "Thursday": 10, "Wednesday": 5}
我如何轻松地做到这一点?
最佳答案
您可以使用.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}