期刊图片复现|Python绘制分组环形柱状图

# =========================================================================================# ====================================== 1. 环境设置 =======================================# =========================================================================================import matplotlib.pyplot as pltimport numpy as npimport matplotlib.patches as mpatches

# =========================================================================================# ======================================2.颜色库=======================================# =========================================================================================COLOR_SCHEMES = {    1: ['#D9B4A7', '#B0C4DE', '#FDEBD0', '#C8E6C9', '#5E4FA2'],}

# =========================================================================================# ======================================3.绘图函数=======================================# =========================================================================================def draw_radial_bar_chart(datasets, models, values, colors, scheme_id):    #创建画布    fig, ax = plt.subplots(figsize=(11, 11),subplot_kw=dict(polar=True))    ax.set_theta_offset(np.pi / 2)#起始角度    ax.set_theta_direction(-1)#顺时针    group_angle = (2 * np.pi) / num_groups#分组弧度    bar_width = (group_angle * (1 - group_gap)) / num_bars#柱子弧度

        #遍历柱子        for j, val in enumerate(group_vals):            #设置字体动态加粗加大            is_mofo = (j == 4)            fontsize = 24 if is_mofo else 16            fontweight = 'bold' if is_mofo else 'normal'            #文本径向坐标            text_r = inner_radius + bar_height - (0.55 if is_mofo else 0.45)            #柱子数值标注            ax.text(theta,  #角度                    text_r,  #半径                    f'{val:.3f}',  #文本                    zorder=4)  #层

        arc_start = group_start_angle#圆心分界弧线的起始弧度        arc_end = group_start_angle + num_bars * bar_width#结束弧度        arc_angles = np.linspace(arc_start, arc_end, 50)#在起始和结束弧度之间均匀生成数据点，用以绘制一条平滑的弧线        ax.text(mid_angle,  #角度                inner_radius - 0.35,  #半径                fontsize=15,  #字体大小                fontweight='bold')  #加粗

    handles = [mpatches.Patch(color=colors[i], label=models[i]) for i in range(len(models))]  #创建图例句柄    #绘制图例    legend = ax.legend(handles=handles,  #句柄                       loc='center',  #坐标                       frameon=True,  #边框                       fontsize=13,  #字体大小                       bbox_to_anchor=(0.5, 0.5),  #精确坐标                       edgecolor='lightgrey')  #边框颜色    #设置图例文本    for text in legend.get_texts():        if text.get_text() == 'MoFo (Ours)':            text.set_color('#C00000')            text.set_weight('bold')

# =========================================================================================# ======================================4.执行部分=======================================# =========================================================================================if __name__ == '__main__':    excel_filename = r"data.xlsx"#原始数据    #设置是否批量绘图    plot_all = True    if plot_all:        for scheme_id in COLOR_SCHEMES.keys():            print('正在绘制并保存方案：', scheme_id)            draw_radial_bar_chart(datasets, models, values, COLOR_SCHEMES[scheme_id], scheme_id)    else:        draw_radial_bar_chart(datasets, models, values, COLOR_SCHEMES[20], 20)

excel_filename = r"data.xlsx"#原始数据

2.设置是否批量绘图，执行部分：

plot_all = True

plt.savefig(rf'scheme_{scheme_id}.png', dpi=300, bbox_inches='tight')