可视化时间序列时标记特定日期

我有一个包含几年数据的时间序列，例如：

ts = pd.Series(np.random.randn(1000), index=pd.date_range('1/1/2000', periods=1000))

ts = ts.cumsum()

ts.plot()

我还有两个额外的数组：让我们调用第一个

dates = [pd.datetime("2000-12-01"), pd.datetime("2001-01-03")]

还有第二个

labels = ["My birthday", "My dad's birthday"]

labels[i] 包含日期[i] 的标签。我想做的是将它们显示在时间序列图中，以便可以识别它们。一种可能的可视化可能是在 x 轴上显示日期，从那里开始绘制一条垂直线，并将标签放在图例中（带有颜色编码）或该线旁边的某处。

最终结果应该与此没有太大不同：

在 pandas 和 matplotlib API 之间切换一开始可能会令人困惑。

解决方案：获取当前轴，然后使用标准matplotlib API进行注释。这让你开始：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

ts = pd.Series(np.random.randn(1000),
               index=pd.date_range('1/1/2000',
               periods=1000))

ts = ts.cumsum()
ts.plot()

label_list = [
    (pd.to_datetime("2001-05-01"), 'My\nbirthday', 'r'),
    (pd.to_datetime("2001-10-16"), "Dad's\nbirthday", 'b')
]

ax = plt.gca()

for date_point, label, clr in label_list:
    plt.axvline(x=date_point, color=clr)
    plt.text(date_point, ax.get_ylim()[1]-4, label,
             horizontalalignment='center',
             verticalalignment='center',
             color=clr,
             bbox=dict(facecolor='white', alpha=0.9))

plt.show()

这会产生下图，您需要考虑修改titles http://matplotlib.org/examples/pylab_examples/figure_title.html, and 文本标签及其边界框 http://matplotlib.org/users/annotations_intro.html到轴对象：