python - 在 Matplotlib 中注释时间序列图

Question

我有一个日期索引数组（x）（日期时间对象）和一个实际值数组（y：债券价格）。做（在 iPython 中）：

plot(x,y)

生成一个非常精细的时间序列图，其中 x 轴标有日期。到目前为止没有问题。但我想在某些日期添加文字。例如，在 2009 年 10 月 31 日，我希望显示文本“事件 1”，并带有指向该日期y值的箭头。

我已经阅读了有关text()和annotate()的 Matplotlib 文档，但无济于事。它仅涵盖标准编号的 x 轴，我无法推断如何在我的问题上处理这些示例。

谢谢

Answer 1

Matplotlib 对日期使用内部浮点格式。

您只需要将日期转换为该格式（使用matplotlib.dates.date2numor matplotlib.dates.datestr2num），然后annotate照常使用。

作为一个有点过于花哨的例子：

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

x = [dt.datetime(2009, 05, 01), dt.datetime(2010, 06, 01), 
     dt.datetime(2011, 04, 01), dt.datetime(2012, 06, 01)]
y = [1, 3, 2, 5]

fig, ax = plt.subplots()
ax.plot_date(x, y, linestyle='--')

ax.annotate('Test', (mdates.date2num(x[1]), y[1]), xytext=(15, 15), 
            textcoords='offset points', arrowprops=dict(arrowstyle='-|>'))

fig.autofmt_xdate()
plt.show()