Pandas制作加权饼图的实现方法

使用Pandas制作加权饼图的具体实现步骤:

  1. 对DataFrame进行groupby分组,并求和:
pythonCopy codenew_df = df.groupby(['color'])[['weight']].sum()

这将统计每个color的权重总和。

  1. 使用reset_index()将索引还原为列:
pythonCopy codenew_df = new_df.reset_index()
  1. 重命名列名称,作为值和名称字段:
pythonCopy codenew_df.columns = ['color', 'total']
  1. 使用plotly express制作饼图:
pythonCopy codefig = px.pie(new_df, values='total', names='color')

指定total列为值,color列为名称。

这样通过分组求和、reset_index以及列重命名,可以轻松地将DataFrame转换为饼图所需的格式。

plotly express能够以更直观的方式展示结果。

Pandas结合plotly提供了制作各类图表的高效便捷实现。