如何在 matplotlib 中生成这些水平线？

Question

Matplotlib 图片我正在尝试添加水平线，如图中的那样，周期长度为 5。我该如何做呢？

Answer 1

黑色阶跃函数可以用 绘制plt.step()。红色和蓝色水平线plt.hlines()：

import matplotlib.pyplot as plt

import numpy as np

N = 150

x = np.random.uniform(0.1, 2, N).cumsum()

y = np.random.normal(0.01, 0.2, N).cumsum()

plt.step(x, y, color='black')

for color in ('red', 'blue'):

    xc = np.linspace(x[0], x[-1], N) + np.random.normal(0, 0.1, N)

    yc = np.interp(xc, x, y) + np.random.normal(0, 0.1, N)

    plt.hlines(yc, xc, xc + 5, colors=color, lw=0.5)

plt.show()