如何基于pandas绘制每年燃料类型占比的百分比堆叠柱状图
解决方案
你当前得到的fuel_percentage是pandas多级索引Series,只需做简单的格式转换就可以直接用于绘图,完整操作步骤如下:
1. 数据格式转换
先把计算好的占比数据转为适合绘图的宽表格式,同时补全1997-2021的所有年份、以及缺失的混合动力(Hybrid)占比数据:
import pandas as pd import matplotlib.pyplot as plt # 将多级索引Series转为DataFrame,重置索引 fuel_df = fuel_percentage.reset_index(name='percentage') # 转换为宽表结构:行=年份,列=燃油类型,值=占比,缺失占比默认补0 fuel_wide = fuel_df.pivot(index='year', columns='fuelType', values='percentage').fillna(0) # 补全1997-2021所有年份,无数据的年份占比全部设为0 all_years = pd.RangeIndex(start=1997, stop=2022, step=1) fuel_wide = fuel_wide.reindex(all_years, fill_value=0) # 确保仅保留需要的三类燃油类型,无对应类型的列自动补0 target_fuel_types = ['Petrol', 'Diesel', 'Hybrid'] for fuel_type in target_fuel_types: if fuel_type not in fuel_wide.columns: fuel_wide[fuel_type] = 0 fuel_wide = fuel_wide[target_fuel_types]
2. 绘制堆叠柱状图
用处理好的宽表直接绘制符合要求的堆叠柱状图:
# 设置画布尺寸 plt.figure(figsize=(16, 7)) # 绘制堆叠柱状图 fuel_wide.plot(kind='bar', stacked=True, ax=plt.gca(), color=['#1f77b4', '#ff7f0e', '#2ca02c']) # 配置图表参数 plt.xlabel('年份', fontsize=12) plt.ylabel('占比(%)', fontsize=12) plt.ylim(0, 100) # 固定Y轴范围为0-100 plt.title('1997-2021年各燃油类型车辆年度占比', fontsize=14) plt.legend(title='燃油类型', labels=['汽油', '柴油', '混合动力'], fontsize=10) # 调整X轴标签避免重叠 plt.xticks(rotation=45, ha='right') plt.tight_layout() plt.show()
如果需要更美观的绘图样式,可以在绘图代码前添加plt.style.use('seaborn-v0_8')调用预设风格。
内容的提问来源于stack exchange,提问作者JacobMarlo
相关产品推荐
相关产品推荐

