期刊图片复现|Python绘制高颜值星状图

# =========================================================================================# ====================================== 1. 环境设置 =======================================# =========================================================================================import matplotlib.pyplot as pltimport numpy as np

# =========================================================================================# ======================================2.颜色库=======================================# =========================================================================================COLOR_SCHEMES = {    1: ['#314B76', '#FC9F5B', '#F3CC71', '#ED6D4E', '#008784', '#144249'],}

# =========================================================================================# ======================================3.绘图函数=======================================# =========================================================================================def plot_advanced_forest_chart(df, scheme_id):    total_plots = len(df) + 1 #子图数    cols = int(np.ceil(np.sqrt(total_plots))) #列数    rows = int(np.ceil(total_plots / cols)) #行数

    outer_circle_linewidth = 2.0 #外圈线宽    max_radius_limit = 1.2 #最大半径    #创建画布    fig, axs = plt.subplots(rows, #行                            cols, #列                            subplot_kw={'projection': 'polar'}, #极坐标图                            figsize=(cols * 10 / 3, rows * 10 / 3)) #画布大小            ax.grid(False) #关闭网格线            ax.spines['polar'].set_visible(True) #外圈边框            ax.spines['polar'].set_edgecolor('black') #外圈边框颜色            ax.spines['polar'].set_linewidth(outer_circle_linewidth) #外圈线宽            ax.set_ylim(0, max_radius_limit) #半径显示范围

    ax_leg = axs[0, 0] #取第一个子图作图例    #设置图例    ax_leg.bar(angles, #角度               [1] * num_cats, #半径               width=width, #扇形宽度               bottom=0.0, #起始半径               color=colors_to_use, #扇形颜色               edgecolor='black', #边框颜色               linewidth=1) #边框线宽

    positions = [] #初始化子图位置列表            if vals[i] > 0:                ax.bar(angles[i], #角度                       vals[i], #半径                       width=width, #宽度                       bottom=0.0, #起始半径                       color=colors_to_use[i], #颜色                       edgecolor='black', #边框颜色                       linewidth=1) #边框线宽

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

excel_file = r'data.xlsx'

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

plot_all = True

categories = [col for col in df.columns if col != 'Bundle'] #类别名

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

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