web-dev-qa-db-ja.com

cオプションで指定された色ラベルと凡例をもつmatplotlib散布図

作成したい この種類の散布図 ここで、ポイントには「c」オプションで指定された色があり、凡例は色の意味を示しています。

私のデータソースは次のようなものです：

scatter_x = [1,2,3,4,5]
scatter_y = [5,4,3,2,1]
group = [1,3,2,1,3] # each (x,y) belongs to the group 1, 2, or 3.

私はこれを試しました：

plt.scatter(scatter_x, scatter_y, c=group, label=group)
plt.legend()

残念ながら、期待通りの伝説は得られませんでした。凡例を適切に表示する方法は？ 5つの行があり、各行に色とグループの対応が表示されていると思いました。

enter image description here

pythonmatplotlibplotlegend
6
rkjt50r983

前述の例のように、各グループに対してplt.scatterを呼び出します。

import numpy as np
from matplotlib import pyplot as plt

scatter_x = np.array([1,2,3,4,5])
scatter_y = np.array([5,4,3,2,1])
group = np.array([1,3,2,1,3])
cdict = {1: 'red', 2: 'blue', 3: 'green'}

fig, ax = plt.subplots()
for g in np.unique(group):
    ix = np.where(group == g)
    ax.scatter(scatter_x[ix], scatter_y[ix], c = cdict[g], label = g, s = 100)
ax.legend()
plt.show()

enter image description here

12
p-robot

これをチェックしてください：

import matplotlib.pyplot as plt
import numpy as  np

fig, ax = plt.subplots()
scatter_x = np.array([1,2,3,4,5])
scatter_y = np.array([5,4,3,2,1])
group = np.array([1,3,2,1,3])
for g in np.unique(group):
    i = np.where(group == g)
    ax.scatter(scatter_x[i], scatter_y[i], label=g)
ax.legend()
plt.show()
3
HISI