如何在Matplotlib中绘制与缩放无关、带箭头头部且支持区域高亮的水平条形图？

我完全懂你遇到的糟心事——辛辛苦苦画的基因组区域箭头图，在预览窗口里看着好好的，一用高DPI保存成图片，箭头头就和条形错位了！而且你还试过FancyArrow，结果因为像素对齐问题导致高亮区域和箭头总有缝隙，确实头疼。咱们来解决这个问题。

问题根源 #

你之前通过像素转换计算箭头长度的方法，其实依赖于当前绘图窗口的DPI参数，但调用fig.savefig()时你指定了更高的DPI，这就导致像素转换的比例和实际保存时的比例不匹配，箭头头自然就歪了。而且这种像素级的转换本来就容易受渲染环境影响，不够稳健。

解决方案：基于轴比例计算箭头长度 #

我们换一种思路，不通过像素，而是基于轴的数据比例来计算箭头长度，让箭头头的长度始终和条形高度保持固定比例，这样不管是预览还是高分辨率保存，比例都不会乱。

修改后的代码如下：

import numpy as np
import matplotlib.patches as patches
import matplotlib.pyplot as plt

def test(bar_height=0.8, figsize=(10, 6), arrow_headlen=0.2, dpi=600):
    X0 = np.arange(0, 10, 1)
    X1 = X0 + 2
    Y = np.arange(0, 10, 1)

    fig = plt.figure(figsize=figsize)
    ax = fig.add_subplot(111)

    # 基于轴的数据比例计算箭头长度：让箭头长度是条形高度的arrow_headlen倍
    # get_data_ratio()返回x轴范围/y轴范围的比例，用来转换y方向长度到x方向
    arrowlen = bar_height * arrow_headlen * ax.get_data_ratio()

    # 绘制灰色条形（箭头主体）
    ax.barh(y=Y, left=X0, width=X1 - X0, height=bar_height, color="0.5")

    # 绘制箭头头部
    for y, x1 in zip(Y, X1):
        yl = y - bar_height / 2  # 条形下边缘
        yh = y + bar_height / 2  # 条形上边缘
        # 绘制三角形箭头头，和条形边缘完全对齐
        arrow = patches.Polygon([(x1, yl), (x1, yh), (x1 + arrowlen, y)], color="0.5")
        ax.add_patch(arrow)

    # 绘制高亮区域（蓝色部分）
    ax.barh(y=Y, left=X0 + 0.5, width=(X1 - X0) / 2, height=bar_height, color="blue")

    # 调整轴范围，避免箭头头被截断
    ax.set_xlim(ax.get_xlim()[0], ax.get_xlim()[1] + arrowlen)

    fig.savefig("./test_from_savefig.png", dpi=dpi)
    plt.show()

test()

关键改动说明 #

• 箭头长度计算：用ax.get_data_ratio()获取轴的x/y数据比例，把条形高度的arrow_headlen倍转换为x轴方向的长度，这样箭头头的大小始终和条形高度成比例，不受DPI或x轴缩放影响。
• 条形边缘计算：把原来的0.49 * bar_height改成bar_height / 2，更精准地对齐条形边缘，避免不必要的像素偏移。
• 轴范围调整：添加了ax.set_xlim()来扩展x轴范围，确保箭头头不会被截断。

效果验证 #

现在不管是用plt.show()预览后保存，还是直接用fig.savefig()高DPI保存，箭头头都会和条形完美对齐，而且高亮区域也不会出现缝隙——完全符合你想要的效果：

• 灰色箭头的头部宽度和主体宽度一致
• 箭头头长度不随x轴缩放变化
• 高亮区域和箭头主体完美融合

备注：内容来源于stack exchange，提问作者mRcSchwering