在几个图上绘制一条线

Question

我不知道这个东西是如何调用的，甚至不知道如何描述它，所以标题可能会有点误导。

第一个附图是使用pyplot创建的。我想绘制一条贯穿所有图形的直线，而不是我目前使用的三个红点。在pyplot中有可能吗？第二个图像是我正在寻找的。

Answer 1

您可以通过关闭相关线路的剪裁来解除此问题。可能有一种更简洁的方法 - 你可能可以直接在主框架上画线 - 但以下对我有用：

from matplotlib import pyplot as plt
from numpy import arange, sin, cos

xx = arange(100)
cut = (xx > 0) & (xx % 17 == 0)
y1 = sin(xx)
y2 = (xx**2) % 2.0+cos(xx+0.5)

fig = plt.figure()
ax1 = fig.add_subplot(211)
ax1.plot(xx, y1, c="blue",zorder=1)
ax1.scatter(xx[cut], y1[cut], c="red",zorder=2)
ax2 = fig.add_subplot(212)
ax2.plot(xx, y2, c="green",zorder=1)
ax2.scatter(xx[cut], y2[cut], c="red",zorder=2)

for x in xx[cut]:
    ax1.axvline(x=x,ymin=-1.2,ymax=1,c="red",linewidth=2,zorder=0, clip_on=False)
    ax2.axvline(x=x,ymin=0,ymax=1.2,c="red",linewidth=2, zorder=0,clip_on=False)

plt.draw()
fig.savefig('pic.png')

通过更多工作，您可以修改线条图来处理多个子图窗口的一般情况，但我非常懒惰。 ：^）

Answer 2

相关文件：
http://matplotlib.sourceforge.net/api/pyplot_api.html#matplotlib.pyplot.axvline

编辑：因为@DSM's answer比我的好得多，所以我羞愧地将一些答案合并在一起，试图让我的答案变得不那么糟糕。

我试图处理列中多个子图的一般情况（即，不是更多一般的多个子图的情况，例如在网格中）。

感谢@DSM，您的回答和@Artium的问题。

import matplotlib.pyplot as plt
import numpy as np

def main():
    fig = plt.figure() 

    x = np.arange(20)
    y1 = np.cos(x)
    y2 = (x**2)
    y3 = (x**3)
    yn = (y1,y2,y3)
    cut = (x > 0) & (x % 2 == 0)
    COLORS = ('b','g','k')

    for i,y in enumerate(yn):
        ax = fig.add_subplot(len(yn),1,i+1)

        ax.plot(x, y,ls='solid', color=COLORS[i], zorder=1) 
        ax.scatter(x[cut], y[cut], c='r', zorder=2)

        if i != len(yn) - 1:
            ax.set_xticklabels( () )

        for j in x[cut]:
            if i != len(yn) - 1:
                ax.axvline(x=j, ymin=-1.2, ymax=1,
                           c='r', lw=2, zorder=0, clip_on=False)
            else:
                ax.axvline(x=j, ymin=0, ymax=1,
                           c='r', lw=2, zorder=0, clip_on=False)

    fig.suptitle('Matplotlib Vertical Line Example')
    plt.show()

if __name__ == '__main__':
    main()

Answer 3

[2013年3月更新] 在较新版本的matplotlib中，ConnectionPatch大大简化了此任务。只要需要覆盖两个以上的子图，它就特别有用。

from matplotlib import pyplot as plt
from matplotlib.patches import ConnectionPatch
from numpy import arange, sin, cos

xx = arange(100)
cut = (xx > 0) & (xx % 17 == 0)
y1 = sin(xx)
y2 = (xx**2) % 2.0+cos(xx+0.5)

fig = plt.figure()
ax1 = fig.add_subplot(211)
ax1.plot(xx, y1, c="blue")
ax1.scatter(xx[cut], y1[cut], c="red")
ax2 = fig.add_subplot(212)
ax2.plot(xx, y2, c="green")
ax2.scatter(xx[cut], y2[cut], c="red")

for x in xx[cut]:
    con = ConnectionPatch(xyA=(x, -1.5), xyB=(x, 1.5),
        coordsA="data", coordsB="data", axesA=ax2, axesB=ax1,
        arrowstyle="-", linewidth=2, color="red")
    ax2.add_artist(con)

plt.draw()
fig.savefig('pic.png')

Answer 4

我会尝试axvline(x, y1, y2)（link），但我不认为pyplot中的任何选项都会绘制跨越多个子图/图的内容。

如果是这种情况，我会尝试在图表中的每个点绘制相同的垂直线，希望将相同的意图传达给观众。