Come disegnare linee verticali su un dato diagramma in matplotlib?

Dato un diagramma del segnale nella rappresentazione del tempo, come disegnare linee che segnano l'indice temporale corrispondente?

In particolare, dato un diagramma di segnale con un indice temporale compreso tra 0 e 2,6 (s), desidero tracciare linee rosse verticali che indicano un indice temporale corrispondente per l'elenco [0.22058956, 0.33088437, 2.20589566], come posso fare?

Il modo standard per aggiungere linee verticali che copriranno l'intera finestra del diagramma senza che tu debba specificare la loro altezza effettiva è plt.axvline

import matplotlib.pyplot as plt

plt.axvline(x=0.22058956)
plt.axvline(x=0.33088437)
plt.axvline(x=2.20589566)

O

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

È possibile utilizzare molte delle parole chiave disponibili per altri comandi trama (ad esempio color, linestyle, linewidth...). Puoi passare gli argomenti delle parole chiave ymine, ymaxse ti piace, gli assi corrdinati (ad esempio ymin=0.25, ymax=0.75copriranno la metà centrale della trama). Esistono funzioni corrispondenti per linee orizzontali ( axhline) e rettangoli ( axvspan).

Per più righe

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

Se qualcuno vuole aggiungere un legende / o colorsalcune linee verticali, usa questo:

import matplotlib.pyplot as plt

# x coordinates for the lines
xcoords = [0.1, 0.3, 0.5]
# colors for the lines
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()

risultati:

Chiamare axvline in un ciclo, come altri hanno suggerito, funziona, ma può essere scomodo perché

1. Ogni linea è un oggetto grafico separato, che rende le cose molto lente quando si hanno molte linee.
2. Quando crei la legenda ogni riga ha una nuova voce, che potrebbe non essere quella che desideri.

Invece puoi usare le seguenti funzioni di convenienza che creano tutte le linee come un singolo oggetto grafico:

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

Oltre a plt.axvlinee plt.plot((x1, x2), (y1, y2))OR plt.plot([x1, x2], [y1, y2])come indicato nelle risposte sopra, si può anche usare

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

per tracciare una linea verticale x_posche va dal y1verso y2in cui i valori y1e y2sono in coordinate assolute dati.