matplotlib 特定のラベルのプロットだけを出力方法

Question

###前提・実現したいこと
前回質問と被ってしまうのですが、特定のラベルのプロットのみ出力する方法を探しています。
（例)下記コードでいうと、LAVEL1とLAVEL3の青と緑だけ散布図として出力したいです。

###該当のソースコード

colors=['blue','black','green','greenyellow','gold','fuchsia','crimson','teal','tomato','aqua','white']
cmap = ListedColormap(colors)

users = np.genfromtxt('/home/srl/mimamori1/b1/12pca.csv', dtype=None, delimiter=",")

pca = PCA(n_components=4)
users_r = pca.fit_transform(users)

color=[1]*402+[2]*150+[3]*360+[4]*36+[5]*42+[6]*42+[7]*48+[8]*783+[9]*118+[10]*30+[11]*2011


plt.subplot(2,3,1)
plt.scatter(users_r[:,0],users_r[:,1] ,c=color,cmap=cmap)
plt.xlabel("pc1")
plt.ylabel("pc2")
plt.legend()
plt.title("test1")

#~~~~~~~~省略~~~~~~~~~~~~~

plt.subplot(2,3,6)
plt.scatter(users_r[:,2],users_r[:,3],c=color,cmap=cmap)
plt.xlabel("pc3")
plt.ylabel("pc4")
plt.legend()
plt.title("test6")

plt.show()

###試したこと
一つのラベルの要素だけ出力はできるようになってたのですが、複数のラベルを選んで出力が如何せんできていません。

Answer 1

まず、color は前回までのように numpy.ndarray型にしておきましょう

color = np.array([1]*402+[2]*150+[3]*360+[4]*36+[5]*42+[6]*42+[7]*48+[8]*783+[9]*118+[10]*30+[11]*2011)

で、例えば 1と3のデータを選択したい場合は

selected = np.isin(color, [1, 3])

とすると、選択された数値の部分が True、その他の部分が False とした Boolean配列が得られますので
あとは、描写の部分を

plt.scatter(users_r[selected,0],users_r[selected,1] ,c=color[selected],cmap=cmap)

と修正することで、選択したデータだけでグラフを書くことができます。

ただこの場合、colormapが範囲が自動調節され、本来の色でプロットできないので、数値の上限・下限を設定して

plt.scatter(users_r[selected,0],users_r[selected,1] ,c=color[selected],cmap=cmap, vmin=1, vmax=11)

とすると良いでしょう。

Comments

[tetuwan28emon]

前回に続き希望通りの結果を得ることができ、今のプロジェクトにもすぐに実装できました。本当にありがとうございます。