Matplotlibでの自動凡例作成

Matplotlibでの凡例の作成を自動化するために、次の手順を実行できます-

• 図のサイズを設定し、サブプロット間およびサブプロットの周囲のパディングを調整します。
• 変数を初期化します、 N、 サンプルデータの数。
• x、y、cを作成します およびs numpyを使用したデータ。
• subplots（）を使用して、図とサブプロットのセットを作成します メソッド。
• xをプロットします およびy さまざまな色とサイズのデータ​​ポイント。
• 軸に凡例を配置します。
• アーティストを追加します 図に。
• PathCollectionの凡例ハンドルとラベルを作成します。
• ここでも、サイズの軸に凡例を配置します。
• 図を表示するには、 show（）を使用します メソッド。

例

import matplotlib.pyplot as plt
import numpy as np

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

N = 45
x, y = np.random.rand(2, N)
c = np.random.randint(1, 5, size=N)
s = np.random.randint(10, 220, size=N)
fig, ax = plt.subplots()
scatter = ax.scatter(x, y, c=c, s=s)
legend1 = ax.legend(*scatter.legend_elements(), loc="lower left", title="Classes")
ax.add_artist(legend1)
handles, labels = scatter.legend_elements(prop="sizes", alpha=0.6)
legend2 = ax.legend(handles, labels, loc="upper right", title="Sizes")

plt.show()

出力