【Python】k-means法におけるクラスタのグラフに凡例や番号をつけたいです！

Question

■使用言語とライブラリ
・Python
・Google Colaboratory

■本文
現在Python実践データ分析100本ノックという本で学習しているのですが、
第4章（34ノック目）の中で、k-means法をつかったクラスタの可視化をしたのですが、
以下のコードで出力される色分けされた散布図にどれがどのクラスタ分類かグラフ内に記述したいのですが、plt.scatter内にlabel = iを入れたり、plt.legend()といれても表示されません。

こちら、グラフ内に、凡例を追加する方法はありますでしょうか？

import matplotlib.pyplot as plt
%matplotlib inline

#customer_clusetering["cluster"]はクラスタの分類を指すカラムです
for i in customer_clustering["cluster"].unique():
  #クラスタの分類（今回は4つ）ごとにテーブルを抽出
  tmp = pca_df.loc[pca_df["cluster"] == i]
  #そのテーブルのカラムで、散布図を作る
  plt.scatter(tmp[0], tmp[1])

なにかおわかりになる方がいらっしゃれば、
ご教示いただけますと幸いです。

Answer 1

Matplotlib の 2D 散布図に凡例を追加するに書かれているようにやってください。

また、labal=には文字列を指定してください。

Comments

[data_kazu]

ppaulさんいつもありがとうございます！こちらできました！ありがとうございます！