Pandasの時系列プロットに垂直線を描画する方法

Pandasを使ってデータフレームを作成し、Matplotlibのaxvline()メソッドを利用することで、時系列プロット上の任意の位置に垂直線を簡単に追加できます。本記事では、その具体的な手順をサンプルコード付きで解説します。

手順

• Pandasを使用して、日付をインデックスとしたデータフレームを作成します。

• データフレームを基に、Matplotlibで時系列の折れ線グラフをプロットします。

• axvline()メソッドを使い、指定した日付の位置に緑色の破線（linestyle="dashed"）で垂直線を追加します。

• 同様にaxvline()を使い、別の日付の位置に赤色の破線で垂直線を追加します。

• plt.show()を実行して、完成したグラフを表示します。

サンプルコード

import pandas as pd
from matplotlib import pyplot as plt
df = pd.DataFrame(index=pd.date_range("2019-07-01", "2019-07-31"))
df["y"] = 1
ax = df.plot()
ax.axvline("2019-07-24", color="green", linestyle="dashed")
ax.axvline("2019-07-31", color="red", linestyle="dashed")
plt.show()

実行結果

上記のコードを実行すると、2019年7月1日から7月31日までの時系列プロットが表示され、7月24日の位置に緑色の破線、7月31日の位置に赤色の破線がそれぞれ描画されます。

ポイント解説

• pd.date_range()を使うことで、開始日から終了日までの日付インデックスを自動生成できます。

• axvline()の第1引数には、x軸上の位置（この場合は日付文字列）を指定します。時系列インデックスの場合、日付形式の文字列がそのまま使えるのが便利です。

• colorパラメータで線の色を、linestyleパラメータで線種（実線・破線・点線など）を自由にカスタマイズできます。

• 特定のイベント日や閾値の日付を強調表示したい場合に、この手法は非常に有効です。