如何偏移Matplotlib中重叠的平行阶梯线段？

问题描述 #

使用Matplotlib绘制阶梯图时，多条平行线段相互重叠遮挡，当前实现代码如下：

import matplotlib.pyplot as plt

x = [0, 1, 2, 3, 4, 5]
y1 = [12, 12, 8, 10, 12, 11]
y2 = [11, 11, 8, 9, 11, 11]
y3 = [11, 10, 7, 11, 11, 11]

for y in [y1, y2, y3]:
    plt.step(x, y, lw=3, where='mid')

当前效果为多条阶梯线段在部分区间重叠，下方线段被上方遮挡；期望实现类似地铁线路图的效果，让重叠的线段向上下轻微偏移，互不遮挡。

解决方案 #

1. 手动偏移（快速实现） #

针对已知的重叠区间，给每条曲线添加固定偏移量，适用于数据量小、重叠规律明确的场景：

import matplotlib.pyplot as plt

x = [0, 1, 2, 3, 4, 5]
y1 = [12, 12, 8, 10, 12, 11]
y2 = [11, 11, 8, 9, 11, 11]
y3 = [11, 10, 7, 11, 11, 11]

# 定义每条曲线的偏移量，可根据视觉效果调整
offsets = [0, 0.3, -0.3]
labels = ['曲线1', '曲线2', '曲线3']

for y, offset, label in zip([y1, y2, y3], offsets, labels):
    y_offset = [val + offset for val in y]
    plt.step(x, y_offset, lw=3, where='mid', label=label)

plt.legend()
plt.show()

2. 自动检测重叠区间并偏移（通用方案） #

如果数据量大、重叠情况复杂，可通过算法自动检测重叠区间，动态调整偏移量：

import matplotlib.pyplot as plt
import numpy as np

def offset_overlapping_steps(y_list, offset_amount=0.2):
    # 转换为numpy数组方便批量计算
    y_arrays = [np.array(y) for y in y_list]
    offsetted_ys = []
    
    # 第一条曲线保持原位置
    offsetted_ys.append(y_arrays[0].copy())
    
    for i in range(1, len(y_arrays)):
        current_y = y_arrays[i]
        prev_y = offsetted_ys[i-1]
        # 检测当前曲线与上一条偏移后曲线的重叠区间
        overlap_mask = current_y == prev_y
        # 重叠区间交替添加正负偏移，避免重复堆叠
        offsets = np.where(overlap_mask, offset_amount * ((i % 2) * 2 - 1), 0)
        offsetted_ys.append(current_y + offsets)
    
    return offsetted_ys

x = [0, 1, 2, 3, 4, 5]
y1 = [12, 12, 8, 10, 12, 11]
y2 = [11, 11, 8, 9, 11, 11]
y3 = [11, 10, 7, 11, 11, 11]

offsetted_ys = offset_overlapping_steps([y1, y2, y3])

for idx, y in enumerate(offsetted_ys):
    plt.step(x, y, lw=3, where='mid', label=f'曲线{idx+1}')

plt.legend()
plt.show()

该算法会自动识别重叠的阶梯段，交替添加正负偏移，确保多条线段互不遮挡。

3. 专用工具包推荐 #

如果需要更专业的线路可视化（类似地铁图），可尝试以下工具：

• NetworkX + Matplotlib：适合构建拓扑线路图，支持自定义边的偏移布局，解决重叠问题。
• PyVis：交互式网络可视化工具，自动处理边的重叠，适合复杂线路展示。
• Plotly：提供灵活的线条样式和偏移配置，支持交互式调整视觉效果。

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