如何修复Pandas+Matplotlib绘制染色体子图直方图的两个问题？

下面针对你遇到的两个问题给出具体的修复方案：

问题1：Y轴标签显示位置异常 #

你当前用plt.ylabel()设置Y轴标签，但多子图场景下这个方法只会作用于最后一个子图，导致标签位置偏离预期。推荐用全局标签的方式来统一设置，同时搭配布局调整避免元素重叠：

问题2：单条数据时的TypeError #

当数据集只有一行时，plt.subplots(nrows=1)返回的ax是单个AxesSubplot对象而非数组，直接用ax[i]索引就会报错。我们可以先将单个子图包装成列表，让后续逻辑能统一处理单/多子图场景。

完整修复代码 #

with open(outputdir + '/' + 'hap_size_byVar_'+ soi +'_'+ prefix+'.png', 'wb') as fig_initial:
    # 根据子图数量动态调整画布大小，避免拥挤
    fig, ax = plt.subplots(nrows=len(hap_stats), sharex=True, figsize=(8, 4*len(hap_stats)))
    
    # 处理单条子图的情况：将单个Axes对象转为列表，统一索引逻辑
    if len(hap_stats) == 1:
        ax = [ax]
    
    for i, data in hap_stats.iterrows():
        # 转换数据为整数列表
        data_i = [int(x) for x in data['num_Vars_by_PI'].split(',')]
        ax[i].hist(data_i, label=str(data['CHROM']), alpha=0.5)
        ax[i].legend()
    
    # 设置全局X轴标签
    plt.xlabel('size of the haplotype (number of variants)')
    # 添加全局Y轴标签，通过fig.text精准控制位置
    fig.text(0.04, 0.5, 'frequency of the haplotypes', va='center', rotation='vertical')
    # 调整标题位置，避免被子图遮挡
    plt.suptitle('histogram of size of the haplotype (number of variants)\nfor each chromosome', y=1.02)
    # 自动调整子图间距，防止标签、标题重叠
    plt.tight_layout()
    
    plt.savefig(fig_initial)

额外优化说明 #

• figsize参数根据子图数量动态设置，保证每个子图有足够的显示空间
• tight_layout()自动处理子图间距，避免元素被裁剪或重叠
• suptitle的y参数微调标题位置，防止被顶部子图遮挡

内容的提问来源于stack exchange，提问作者everestial