使用Pandas制作加权饼图的具体实现步骤:
- 对DataFrame进行groupby分组,并求和:
pythonCopy codenew_df = df.groupby(['color'])[['weight']].sum()
这将统计每个color的权重总和。
- 使用reset_index()将索引还原为列:
pythonCopy codenew_df = new_df.reset_index()
- 重命名列名称,作为值和名称字段:
pythonCopy codenew_df.columns = ['color', 'total']
- 使用plotly express制作饼图:
pythonCopy codefig = px.pie(new_df, values='total', names='color')
指定total列为值,color列为名称。
这样通过分组求和、reset_index以及列重命名,可以轻松地将DataFrame转换为饼图所需的格式。
plotly express能够以更直观的方式展示结果。
Pandas结合plotly提供了制作各类图表的高效便捷实现。
标签:
Python