[Matplotlib] 凡例の表示

凡例の設定

matplotlib.axes.Axes.legend

　matplotlib.axes.Axes.legend() は Axes (サブプロット) に 凡例 を添えます。
　凡例とは、Axes.plot() などの描画メソッドで label に渡したテキストのことです。デフォルト設定ではグラフと重ならないように位置を自動調整して表示されます。

# MATPLOTLIB_LEGEND_01-1

import matplotlib.pyplot as plt
import numpy as np

# FigureとAxes
fig = plt.figure(figsize =(4, 4))
ax = fig.add_subplot(111)
ax.grid()

# データを作成
x = np.linspace(-3, 3, 65)
f = x + 3
g = x**2

# データをプロット
ax.plot(x, f, label="y = x**2", color="blue")
ax.plot(x, g, label="y = x**3", color="red")

# 凡例を表示
ax.legend()

　loc オプションで 凡例の位置 (location) を指定できます。
　たとえば、"upper left" を指定すると左上に凡例が表示されます。

# MATPLOTLIB_LEGEND_01-2

# 凡例を左上隅に表示
ax.legend(loc="upper left")

# グラフを再表示
display(fig)

　loc に指定可能な文字列とコードを表にまとめておきます。

　Axes の左下隅を (0, 0)、右上隅を (1, 1) とする float のタプルで位置を指定することもできます。

# MATPLOTLIB_LEGEND_01-3

# 凡例の位置を指定
ax.legend(loc=(0.62, 0.05))

# グラフを再表示
display(fig)

　凡例フレームを表示しないときは frameon に False を指定します。

# MATPLOTLIB_LEGEND_01-4

# 凡例フレームを表示しない
ax.legend(frameon=False)

# グラフを再表示
display(fig)

　凡例フレームの角の丸めは fancybox、枠線の色は edgecolor で指定します。

# MATPLOTLIB_LEGEND_01-5

# 凡例フレームの角を丸めない、枠線を黒にする
ax.legend(fancybox=False, edgecolor="black")

# グラフを再表示
display(fig)

　borderpad は凡例フレームの余白、shadow は影の有無を指定する引数です。

# MATPLOTLIB_LEGEND_01-6

# 凡例の余白を設定、影をつける
ax.legend(borderpad=2, shadow=True)

# グラフを再表示
display(fig)

　ncol で凡例の列数を指定できます。

# MATPLOTLIB_LEGEND_01-7

# 凡例を2列で表示
ax.legend(ncol=2)

# グラフを再表示
display(fig)

　枠線の太さを変えたいときは、get_frame().set_linewidth() メソッドを使います。

# MATPLOTLIB_LEGEND_01-8

# 枠線の太さを設定
ax.legend(edgecolor="black").get_frame().set_linewidth(1.5)

# グラフを再表示
display(fig)

凡例の実践例

　1 枚のサブプロットに sinx と cosx をプロットして 凡例 を添えるサンプルコードです。

# MATPLOTLIB_LEGEND_02

import numpy as np
import matplotlib.pyplot as plt

# FigureとAxes
fig = plt.figure()
ax = fig.add_subplot(111)

# xのデータを作成
x = np.arange(-4, 4, 0.1)

# yのデータを作成
f = np.cos(x)
g = np.sin(x)

# 軸ラベルの設定
ax.set_xlabel("x", size = 12)
ax.set_ylabel("y", size = 12)

# データをプロット
ax.plot(x, f, color = "blue", label = "cosx")
ax.plot(x, g, color = "red", label = "sinx")

# 凡例の設定
ax.legend(loc = "upper right")