期刊图片复现|Python绘制配对云雨图

# =========================================================================================# ====================================== 1. 环境设置 =======================================# =========================================================================================import numpy as npimport pandas as pdimport matplotlib.pyplot as pltfrom matplotlib.patches import Patch

# =========================================================================================# ======================================2.颜色库=======================================# =========================================================================================COLOR_SCHEMES = {    1: ['#F08B94', '#C92B38', '#6E9AB7', '#165785'],}

# =========================================================================================# ======================================3.小提琴图绘制函数=======================================# =========================================================================================def plot_half_violin(ax, #坐标轴                     data,#数据                     pos, #位置                     side='left', #方向                     color_fill='blue',#填充颜色                     color_edge='blue'):  #边缘颜色        #设定保留左半侧        if side == 'left':            vertices[:, 0] = np.clip(vertices[:, 0], -np.inf, pos)  #将顶点X轴坐标限制在负无穷到中心位置之间        else:            vertices[:, 0] = np.clip(vertices[:, 0], pos, np.inf)

# =========================================================================================# ======================================4.箱线图样式设置辅助函数=======================================# =========================================================================================def style_boxplot(bp, edge_color, fill_color):  #箱线图对象、边缘颜色和填充颜色    # 遍历箱线图须线、帽线和中位线    for element in ['whiskers', 'caps', 'medians']:        box.set(color=edge_color,  #边缘颜色                linewidth=1.5)  #边缘线粗细        box.set(facecolor=fill_color)  #填充颜色

# =========================================================================================# ======================================5.主绘图函数=======================================# =========================================================================================def plot_raincloud_chart(df, scheme_id):    colors = COLOR_SCHEMES[scheme_id]  #提取配色方案    for spine in ax.spines.values():        spine.set_linewidth(1.5)    #配置坐标轴刻度    ax.tick_params(axis='both',  #x、y                   which='major',  #主刻度                   length=6,  #长                   width=1.5,  #宽                   labelsize=12)  #字体大小

    base_pre = 1.0  #左侧图x轴基准位置    #调用小提琴图绘制函数    plot_half_violin(ax,                     virginica_pre,                     pos=base_pre - 0.2,                     side='left',                     color_fill=color_virgi_fill,                     color_edge=color_virgi_edge)    ax.scatter(jitter_versi_pre,  #x               versicolor_pre,  #y               facecolors=color_versi_fill,  #填充颜色               edgecolors=color_versi_edge,  #边缘颜色               linewidths=1,  #边缘线宽               alpha=1,  #透明度               s=40,  #大小               zorder=2)  #层    # 样式调整    style_boxplot(bp2,color_virgi_edge,color_virgi_fill)

    base_post = 2.0 #右侧图x轴基准位置    #Virginica右侧箱线图x位置    pos_box_virgi_post = base_post - 0.18    #Versicolor右侧箱线图x位置    pos_box_versi_post = base_post - 0.08    #绘制箱线图    bp4 = ax.boxplot([versicolor_post],  #数据                     positions=[pos_box_versi_post],  #x                     widths=0.06,  #宽                     patch_artist=True, #颜色填充                     showfliers=False)  #离群值    plot_half_violin(ax,virginica_post,pos=base_post + 0.2,side='right', color_fill=color_virgi_fill,color_edge=color_virgi_edge)    #Versicolor右侧小提琴图    plot_half_violin(ax,versicolor_post,pos=base_post + 0.2,side='right',color_fill=color_versi_fill,color_edge=color_versi_edge)

    #Versicolor数据均值    mean_versi_pre, mean_versi_post = np.mean(versicolor_pre), np.mean(versicolor_post)    ax.plot([pos_box_virgi_pre, pos_box_virgi_post], #x            [mean_virgi_pre, mean_virgi_post],#y            color=color_virgi_edge, #颜色            marker='D', #均值点形状            markersize=6,  #点大小            linewidth=1.5, #线宽            zorder=3)  #层

    #X轴刻度线位置    ax.set_xticks([(base_pre + 0.04), (base_post - 0.04)])    # X轴刻度标注    ax.set_xticklabels(['Pre', 'Post'],  #文本                       fontsize=12,  #字体大小                       fontweight='bold')  #加粗    ax.spines['top'].set_visible(False)  #隐藏    ax.spines['right'].set_visible(False)    ax.spines['left'].set_position(('outward', 10))  #移动    ax.spines['bottom'].set_position(('outward', 10))    ax.spines['bottom'].set_bounds((base_pre + 0.04), (base_post - 0.04))  #范围    ax.spines['left'].set_bounds(2.0, 4.5)

    #实例化图例    legend_elements = [    #绘制图例    ax.legend(handles=legend_elements,  #句柄              title='Flower Species',  #标题              loc='upper right',  #位置              bbox_to_anchor=(1.15, 1.05),  #坐标              frameon=False,  #边框              title_fontsize=12,  #标题大小              fontsize=11)  #文本大

# =========================================================================================# ======================================2.颜色库=======================================# =========================================================================================if __name__ == '__main__':    excel_filename = r"data.xlsx"#原始数据    df_real = pd.read_excel(excel_filename)#读取    plot_all = True#是否批量绘图    if plot_all:        for scheme_id in COLOR_SCHEMES.keys():            selected_hex_colors = COLOR_SCHEMES[scheme_id]            print(f'正在绘制并保存配色方案：{scheme_id}')            plot_raincloud_chart(df_real, scheme_id)    else:        scheme_id = 1        selected_hex_colors = COLOR_SCHEMES[scheme_id]        print(f'正在绘制并保存方案：{scheme_id}')        plot_raincloud_chart(df_real, scheme_id)

excel_filename = r"F:\公众号素材\20260612-配对云雨图\simulated_iris_data.xlsx"#原始数据

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

plot_all = True#是否批量绘图

versicolor_pre = df['Versicolor_Pre'].dropna().valuesversicolor_post = df['Versicolor_Post'].dropna().valuesvirginica_pre = df['Virginica_Pre'].dropna().valuesvirginica_post = df['Virginica_Post'].dropna().values

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

plt.savefig(fr"raincloud_plot_scheme_{scheme_id}.pdf", bbox_inches='tight')