In un grafico matplotlib, posso evidenziare intervalli di valori x specifici?

Question

Sto facendo una visualizzazione dei dati di borsa storici per un progetto, e mi piacerebbe evidenziare le regioni di gocce. Ad esempio, quando il titolo sta registrando un drawdown significativo, vorrei evidenziarlo con una regione rossa.

Posso farlo automaticamente o dovrò disegnare un rettangolo o qualcosa del genere?

Answer 1

Dai un'occhiata a axvspan (e axhspan per evidenziare una regione dell'asse y).

import matplotlib.pyplot as plt 

plt.plot(range(10)) 
plt.axvspan(3, 6, color='red', alpha=0.5) 
plt.show() 

Se stai usando date, allora avrete bisogno di convertire i tuoi min e max valori x per matplotlib date. Utilizzare matplotlib.dates.date2num per gli oggetti datetime o matplotlib.dates.datestr2num per vari timestamp di stringa.

import matplotlib.pyplot as plt 
import matplotlib.dates as mdates 
import datetime as dt 

t = mdates.drange(dt.datetime(2011, 10, 15), dt.datetime(2011, 11, 27), 
        dt.timedelta(hours=2)) 
y = np.sin(t) 

fig, ax = plt.subplots() 
ax.plot_date(t, y, 'b-') 
ax.axvspan(*mdates.datestr2num(['10/27/2011', '11/2/2011']), color='red', alpha=0.5) 
fig.autofmt_xdate() 
plt.show()