matplotでラベルの色分けができない

Question

from sklearn.datasets import load_iris
import matplotlib.pyplot as plt
from sklearn.decomposition import PCA

iris = load_iris()

x = PCA(n_components=3).fit_transform(iris.data)

from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure(figsize=(8,6))
ax = Axes3D(fig, elev=-150, azim=110)

# for i in range(3):　　　#これだと、ラベルの色分けができない
#     d = x[iris.target == i, :]   #それぞれの花のデータを種類別で分けｄにいれる。
#     ax.scatter(x[:,0], x[:,1], x[:,2], c=iris.target,label=iris.target_names[i]) #cはカラー分けのプロット

#setosa
ax.scatter(x[:,0], x[:,1], x[:,2],
           label=iris.target_names[0],
           c=iris.target)

# versicolor
ax.scatter(x[:,0], x[:,1], x[:,2],
           label=iris.target_names[0],
           c=iris.target)

# virginica
ax.scatter(x[:,0], x[:,1], x[:,2],
           label=iris.target_names[2],
           c=iris.target)

ax.set_xlabel("1st")
ax.set_ylabel("2nd")
ax.set_zlabel("3rd")
plt.legend()

質問

このプログラムではプロットは色分けできていますが、labelのとこが色分けできていません。どうすれば、いいのでしょうか？

Answer 1

選択が効いていないので、全データで上書きしていることになります。

以下のようにしてみてください。

from sklearn.datasets import load_iris
import matplotlib.pyplot as plt
from sklearn.decomposition import PCA

iris = load_iris()

x = PCA(n_components=3).fit_transform(iris.data)

from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure(figsize=(8,6))
ax = Axes3D(fig, elev=-150, azim=110)

for i in range(3):
    d = x[iris.target == i, :]
    ax.scatter(d[:,0], d[:,1], d[:,2], label=iris.target_names[i])  # dに対して取る


ax.set_xlabel("1st")
ax.set_ylabel("2nd")
ax.set_zlabel("3rd")
plt.legend()
plt.show()

Comments

[cmbnur]

なるほど、上書きさせていたのですね。ありがとうございます。