Matplotlibで複数のポイントにテキスト注釈を付ける方法

Matplotlibで複数のポイントに対して注釈テキストを追加するには、以下の手順に従います。

• 図のサイズを設定し、サブプロット間および周囲のパディングを調整します。
• numpyを使ってxとyのデータポイントを作成します。
• 各散布点にラベルを付けるため、ラベルのリストを作成します。
• scatter()メソッドを使ってxpointsとypointsをプロットします。色にはxpointsを指定します。
• zipでまとめたラベル、xpoints、ypointsをループで反復処理します。
• forループ内でannotate()メソッドを使い、太字のLaTeX形式で注釈を表示します。
• 図を表示するにはshow()メソッドを使用します。

コード例

import numpy as np
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

xpoints = np.linspace(1, 10, 10)
ypoints = np.random.rand(10)

labels = ["%.2f" % i for i in xpoints]

plt.scatter(xpoints, ypoints, c=xpoints)

for label, x, y in zip(labels, xpoints, ypoints):
    plt.annotate(
      f"$\\bf{label}$",
      xy=(x, y), xytext=(-20, 20),
      textcoords='offset points', ha='center', va='bottom',
      arrowprops=dict(arrowstyle='->', connectionstyle='arc3,rad=0'))

plt.show()

出力結果

上記のコードを実行すると、各散布点に矢印付きの太字ラベルが注釈として表示されたグラフが出力されます。注釈はポイントからオフセットした位置（左に20pt、上に20pt）に配置され、矢印によって対応するポイントと結ばれます。