期刊图片复现|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'):  # 边缘颜色    # 绘制小提琴图    v = ax.violinplot(data,  # 数据        # 设定保留左半侧        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']:        # 修改图形元素属性    # 遍历箱体    for box in bp['boxes']:        box.set(color=edge_color,  # 边缘颜色                linewidth=1.5)  # 边缘线粗细        box.set(facecolor=fill_color)  # 填充颜色

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

# =========================================================================================# ======================================5.主绘图函数=======================================# =========================================================================================def plot_raincloud_chart(df, scheme_id):    colors = COLOR_SCHEMES[scheme_id]  # 提取配色方案    color_versi_fill = colors[0]  # Versicolor填充颜色    color_versi_edge = colors[1]  # Versicolor边缘颜色    color_virgi_fill = colors[2]  # Virginica填充颜色    color_virgi_edge = colors[3]  # Virginica边缘颜色    # 配置坐标轴刻度    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_virgi_pre,  # x               virginica_pre,  # y               facecolors=color_virgi_fill,  # 填充颜色               edgecolors=color_virgi_edge,  # 边缘颜色               linewidths=1,  # 边缘线宽               alpha=1,  # 透明度               s=25,  # 大小               zorder=2)  # 层    # Versicolor左侧数据x轴抖动坐标    jitter_versi_pre = np.random.uniform(base_pre - 0.04, base_pre, size=n_samples)    # 样式调整    style_boxplot(bp2, color_virgi_edge, color_virgi_fill)

    base_post = 2.0  # 右侧图x轴基准位置    # Virginica右侧箱线图x位置    pos_box_virgi_post = base_post - 0.18    # 绘制箱线图    bp3 = ax.boxplot([virginica_post],  # 数据                     positions=[pos_box_virgi_post],  # x    # 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)  # 离群值    # Virginica右侧小提琴图    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)    # Virginica数据均值    mean_virgi_pre, mean_virgi_post = np.mean(virginica_pre), np.mean(virginica_post)    #绘制Virginica均值点    ax.plot([pos_box_virgi_pre, pos_box_virgi_post],  #x            [mean_virgi_pre, mean_virgi_post],  #y            color=color_virgi_edge,  #颜色            marker='D',  #均值点形状            linestyle='none',  #连线            markersize=6,  #点大小            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['bottom'].set_bounds((base_pre + 0.04), (base_post - 0.04))  # 范围    ax.spines['left'].set_bounds(2.0, 4.5)

# 实例化图例    legend_elements = [        Patch(facecolor=color_versi_fill,  # 图例色块              edgecolor='none',  # 边框    # 绘制图例              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)  # 读取        selected_hex_colors = COLOR_SCHEMES[scheme_id]        print(f'正在绘制并保存方案：{scheme_id}')        plot_raincloud_chart(df_real, scheme_id)

excel_filename = r"data.xlsx" 

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