期刊图片复现|Python绘制堆叠条形图+弧状条形图组图

# =========================================================================================# ====================================== 1. 环境设置 =======================================# =========================================================================================import matplotlib.pyplot as pltfrom matplotlib.patches import Wedge, Rectangle, Patchimport numpy as npimport pandas as pdimport os

# =========================================================================================# ======================================2.颜色库=======================================# =========================================================================================COLOR_SCHEMES = {    1: ['#ECA8C4', '#314481', '#E9A390', '#9E97C2', '#9BB5D5', '#F8D0C3', '#B4AED5'],}

# =========================================================================================# ======================================4.主绘图函数=======================================# =========================================================================================def plot_advanced_forest_chart(df_real, scheme_id):    ax.set_aspect('equal')  # 强制x和y轴的比例一致    ax.axis('off')  # 去掉坐标轴的刻度和边框显示    ax.set_xlim(-1.2, 1.2)  # x轴的范围    ax.set_ylim(-0.3, 1.4)  # y轴的范围    arc_width = 0.16  # 弧状条形的宽度    gap = 0.02  # 图形之间的间距

    r1_out = 1.05  # 最外侧弧形的外半径    wedge_blue = Wedge(center=(0, 0),  # 中心点                       r=r2_out,  # 外半径                       theta1=theta_blue_start,  # 起始角度                       theta2=180,  # 结束角度                       width=arc_width,  # 圆环宽度                       facecolor=c_arc_blue,  # 填充色                       edgecolor='none')  # 边框线    # 绘制弧形    wedge_purple = Wedge(center=(0, 0),  # 中心点                         r=r4_out,  # 外半径                         theta1=theta_purple_start,  # 起始角度                         theta2=180,  # 结束角度                         width=arc_width,  # 圆环宽度                         facecolor=c_arc_purple,  # 填充色                         edgecolor='none')  # 边框线    ax.add_patch(wedge_purple)  # 添加到坐标轴上

    bar_y = -0.22  # 水平堆叠条形图的y坐标起点    #创建第一部分条形    rect_lb = Rectangle((x1, bar_y),  #坐标                        w_1,  # 宽                        bar_height,  # 高                        facecolor=c_bar_lb,  # 填充色                        edgecolor='none')  # 边框    ax.add_patch(rect_lb)  # 添加到图上    # 创建第二部分条形    rect_pe = Rectangle((x2, bar_y),  #坐标                        w_2,  # 宽                        bar_height,  # 高                        facecolor=c_bar_pe,  # 填充色                        edgecolor='none')  # 边框    ax.add_patch(rect_pe)  # 添加到图上    # 创建第三部分条形    rect_pu = Rectangle((x3, bar_y),  #坐标                        w_3,  # 宽                        bar_height,  # 高                        facecolor=c_bar_pu,  # 填充色                        edgecolor='none')  # 边框    ax.add_patch(rect_pu)  # 添加到图上

    r1_mid = (r1_out + r1_in) / 2  # 第一圈弧形的中径    text_y = bar_y + bar_height / 2  # 底部堆叠条形图内文本的Y轴中间坐标    # 添加第一部分矩形文本    ax.text(x1 + w_1 / 2,  # x            text_y,  # y            texts[4],  # 文本            ha='center',  # 水平            va='center',  # 垂直            color='white', # 颜色            fontsize=16,  # 字体            )    # 添加第二部分矩形文本    ax.text(x2 + w_2 / 2,  # x            text_y,  # y            texts[5],  # 文本            ha='center',  # 水平            va='center',  # 垂直            color='#333333',  # 颜色            fontsize=16,  # 字体            )    # 添加第三部分矩形文本    ax.text(x3 + w_3 / 2,  # x            text_y,  # y            texts[6],  # 文本            ha='center',  # 水平            va='center',  # 垂直            color='white',  # 颜色            fontsize=16,  # 字体            )

    # 主标题    draw_curved_text(ax,  # 轴对象                     "Saitama",  # 文本                     r=1.09,  # 半径                     center_angle=90,  # 中心对齐                     char_spacing=4)  # 间隔    # 创建图例    legend_handles = [        Patch(facecolor=c_arc_pink, label=categories[0]),        Patch(facecolor=c_arc_blue, label=categories[1]),        Patch(facecolor=c_arc_peach, label=categories[2]),        Patch(facecolor=c_arc_purple, label=categories[3]),        Patch(facecolor=c_bar_lb, label=categories[4]),        Patch(facecolor=c_bar_pe, label=categories[5]),        Patch(facecolor=c_bar_pu, label=categories[6])    ]

# =========================================================================================# ======================================4.执行部分=======================================# =========================================================================================if __name__ == '__main__':    excel_path = r'data.xlsx'  #原始数据路径    df_real = pd.read_excel(excel_path)  # 读取数据    scheme_id = 1    selected_hex_colors = COLOR_SCHEMES[scheme_id]    print('正在绘制并保存方案：', scheme_id)    plot_advanced_forest_chart(df_real, scheme_id)

r'data.xlsx'

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

plot_all = True

texts = [f"{int(val)}%" for val in df_real['Value']]categories = df_real['Category'].tolist()

4.设置绘图结果的保存地址，主绘图函数部分：

plt.savefig(fr'Scheme_{scheme_id}.png', bbox_inches='tight')