【Python】Matplotlibでグラフの線にラベルを付ける方法

Matplotlibでグラフの線にラベルを付けるには、plot()メソッドの引数にlabelを指定します。ラベルを付けた線は、legend()メソッドで凡例として図に表示できます。

手順

• 図のサイズを設定し、サブプロット間および周囲の余白を調整します。
• plot()メソッドを使用し、label="line1" を指定して1本目の線をプロットします。
• 同様に、label="line2" を指定して2本目の線をプロットします。
• 図に凡例を配置するには、legend()メソッドを使用します。
• 図を画面に表示するには、show()メソッドを使用します。

コード例

from matplotlib import pyplot as plt

# 図のサイズとレイアウトを設定
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

# ラベル付きでプロット
line1, = plt.plot([1, 2, 3], label="line1")
line2, = plt.plot([3, 2, 1], label="line2")

# 凡例を上部中央に配置
leg = plt.legend(loc='upper center')

plt.show()

実行結果

上記のコードを実行すると、2本の線それぞれに「line1」「line2」というラベルが付いた凡例が、図の上部中央に表示されます。

ポイント

• label引数は、plot()だけでなくscatter()やbar()など他の描画メソッドでも使用できます。
• loc引数には「upper right」「lower left」など、さまざまな位置を指定できます。
• 凡例を図の外側に配置したい場合は、bbox_to_anchor引数を併用すると便利です。