在条形图子图中为每个分布绘制多条水平线Matplotlib

时间:2019-04-07 15:25:32

标签: python python-3.x matplotlib plot

我正在尝试将平均计算值绘制成一条线,穿过我的数据集每个绘制分布的中心。

我的代码如下:

for plot, var in zip(range(1, plot_num+1), var_list):

    ax = fig.add_subplot(2, 2, plot)

    # calculate averages

    sns.stripplot(x=cluster_index_sample[cluster_type], y=cluster_index_sample[var], 
                  jitter=jitter, linewidth=line_width, alpha=alpha, cmap=RS_colorwheel,
                  size=marker_size, ax=ax)

    # create average lines
    ax.axhline(y=cluster_index_sample['Average_'+(var)].iloc[0], 
                                      linewidth=3, xmin=0.2, xmax=0.5)

    ax.set_ylabel(str(var), fontsize=y_lab)
    ax.set_xlabel('')
    ax.tick_params(axis='both', which='major', pad=10)

但是,当我绘制此图时,每个 cluster_type(x轴类别)仅会一次显示水平线。

enter image description here

如何获取它,以便每组带编号的分类值都有各自的平均值?

1 个答案:

答案 0 :(得分:0)

由于您未提供MCVE,因此我无法运行您的代码。不过,您可以尝试使用第二个for循环遍历所有变量以绘制水平平均线,如下所示。您还必须为每行修改xminxmax。我留给你。

for plot, var in zip(range(1, plot_num+1), var_list):
    ax = fig.add_subplot(2, 2, plot)
    sns.stripplot(x=cluster_index_sample[cluster_type], y=cluster_index_sample[var], 
                  jitter=jitter, linewidth=line_width, alpha=alpha, cmap=RS_colorwheel,
                  size=marker_size, ax=ax)
    for v in var_list: # <--- Added here
        ax.axhline(y=cluster_index_sample['Average_'+(v)].iloc[0], 
                                      linewidth=3, xmin=0.2, xmax=0.5) # <--- Added here

    ax.set_ylabel(str(var), fontsize=y_lab)
    ax.set_xlabel('')
    ax.tick_params(axis='both', which='major', pad=10)