Могу ли я выделить определенные диапазоны значений x на графике matplotlib?

Question

Я делаю визуализацию исторических данных о запасах для проекта, и я хотел бы выделить области капель.Например, когда акция испытывает значительную просадку, я хотел бы выделить ее красной областью.

Могу ли я сделать это автоматически, или мне нужно будет нарисовать прямоугольник или что-то еще?

Answer 1

Посмотрите на axvspan (и axhspan для выделения области оси Y).

import matplotlib.pyplot as plt

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

Если выЕсли вы используете даты, то вам нужно будет конвертировать значения min и max x в даты matplotlib.Используйте matplotlib.dates.date2num для datetime объектов или matplotlib.dates.datestr2num для различных строковых временных меток.

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()