首页 文章

如何在matplotlib中给定绘图上绘制垂直线?

提问于
浏览
144

给定时间表示中的信号图,如何绘制标记相应时间指标的线?

具体来说,给定时间索引范围为0到2.6(s)的信号图,我想绘制垂直红线,表示列表 [0.22058956, 0.33088437, 2.20589566] 的相应时间索引,我该怎么办?

5 回答

  • 35

    添加垂直线条以覆盖整个绘图窗口而不必指定其实际高度的标准方法是 plt.axvline

    import matplotlib.pyplot as plt
    
    plt.axvline(x=0.22058956)
    plt.axvline(x=0.33088437)
    plt.axvline(x=2.20589566)
    

    要么

    xcoords = [0.22058956, 0.33088437, 2.20589566]
    for xc in xcoords:
        plt.axvline(x=xc)
    

    您可以使用许多可用于其他绘图命令的关键字(例如 colorlinestylelinewidth ...) . 如果你喜欢轴corrdinates,你可以传递关键字参数 yminymax (例如 ymin=0.25ymax=0.75 将覆盖图的中间一半) . 水平线( axhline )和矩形( axvspan )有相应的功能 .

  • 245

    如果有人想要在某些垂直线上添加图例和/或颜色,请使用:


    import matplotlib.pyplot as plt
    
    xcoords = [0.1, 0.3, 0.5]
    colors = ['r','k','b']
    
    for xc,c in zip(xcoords,colors):
        plt.axvline(x=xc, label='line at x = {}'.format(xc), c=c)
    
    plt.legend()
    plt.show()
    

    Results:

  • 3

    除了上面的答案中提供的 plt.axvlineplt.plot((x1, x2), (y1, y2))plt.plot([x1, x2], [y1, y2]) 之外,还可以使用

    plt.vlines(x_pos, ymin=y1, ymax=y2)
    

    x_posy1y2 绘制一条垂直线 .

  • 5

    正如其他人所建议的那样,在一个循环中调用axvline是有效的,但是可能会带来不便

    • 每一行都是一个单独的绘图对象,当你有很多行时会导致事情变得很慢 .

    • 创建图例时,每一行都有一个新条目,可能不是您想要的 .

    相反,您可以使用以下便捷函数将所有行创建为单个绘图对象:

    import matplotlib.pyplot as plt
    import numpy as np
    
    
    def axhlines(ys, ax=None, **plot_kwargs):
        """
        Draw horizontal lines across plot
        :param ys: A scalar, list, or 1D array of vertical offsets
        :param ax: The axis (or none to use gca)
        :param plot_kwargs: Keyword arguments to be passed to plot
        :return: The plot object corresponding to the lines.
        """
        if ax is None:
            ax = plt.gca()
        ys = np.array((ys, ) if np.isscalar(ys) else ys, copy=False)
        lims = ax.get_xlim()
        y_points = np.repeat(ys[:, None], repeats=3, axis=1).flatten()
        x_points = np.repeat(np.array(lims + (np.nan, ))[None, :], repeats=len(ys), axis=0).flatten()
        plot = ax.plot(x_points, y_points, scalex = False, **plot_kwargs)
        return plot
    
    
    def axvlines(xs, ax=None, **plot_kwargs):
        """
        Draw vertical lines on plot
        :param xs: A scalar, list, or 1D array of horizontal offsets
        :param ax: The axis (or none to use gca)
        :param plot_kwargs: Keyword arguments to be passed to plot
        :return: The plot object corresponding to the lines.
        """
        if ax is None:
            ax = plt.gca()
        xs = np.array((xs, ) if np.isscalar(xs) else xs, copy=False)
        lims = ax.get_ylim()
        x_points = np.repeat(xs[:, None], repeats=3, axis=1).flatten()
        y_points = np.repeat(np.array(lims + (np.nan, ))[None, :], repeats=len(xs), axis=0).flatten()
        plot = ax.plot(x_points, y_points, scaley = False, **plot_kwargs)
        return plot
    
  • 15

    对于多行

    xposition = [0.3, 0.4, 0.45]
    for xc in xposition:
        plt.axvline(x=xc, color='k', linestyle='--')
    

相关问题