クラスタ毎の色変更ができません

Question

###前提・実現したいこと
クラスタリングの勉強をしています．
scikit-learnでクラスタリングしようと思っています．
下記の２つの記事を参考にさせていただいています．
こちらの記事では，棒グラフですが，散布図に変更しようとしています．
上手く図は出たのですが，色が一色だけしか表示できず，
クラスタ毎に色を変更したいと思います．
よろしくお願い致します．

###発生している問題・エラーメッセージ

散布図は表示されるのですが，青一色だけでクラスタ毎に色を変えたいと思います．

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

#!/usr/bin/env python
# -*- coding: utf8 -*-

import pandas as pd
import numpy as np
from sklearn.cluster import KMeans
import matplotlib.pyplot as plt

#クラスタ数
N_CLUSTERS = 5
# データセットを読み込み
cust_df = pd.read_csv("Wholesale_customers_data.csv")

# 不要なカラムを削除
del(cust_df['Channel'])
del(cust_df['Region'])

#print(cust_df)

# Pandas のデータフレームから Numpy の行列 (Array) に変換
cust_array = np.array([cust_df['Fresh'].tolist(),
                       cust_df['Milk'].tolist(),
                       cust_df['Grocery'].tolist(),
                       cust_df['Frozen'].tolist(),
                       cust_df['Milk'].tolist(),
                       cust_df['Detergents_Paper'].tolist(),
                       cust_df['Delicassen'].tolist()
                       ], np.int32)

# 行列を転置
#cust_array = cust_array.T
features = cust_array.T

# クラスタ分析を実行 (クラスタ数=4)
#pred = KMeans(n_clusters=4).fit_predict(cust_array)
cls = KMeans(n_clusters=N_CLUSTERS)
pred = cls.fit_predict(features)
#クラスタ後，ラベル表示
print(pred)

# Pandas のデータフレームにクラスタ番号を追加
#cust_df['cluster_id']=pred


# 可視化（散布図）
for i in range(N_CLUSTERS):
    labels = features[pred == i]
    plt.scatter(labels[:, 0], labels[:, 1])

# クラスタのセントロイド (重心) を描く
centers = cls.cluster_centers_
plt.scatter(centers[:, 0], centers[:, 1], s=100,
                facecolors='none', edgecolors='black')

plt.show()

###試したこと
参考記事
http://blog.amedama.jp/entry/2017/03/19/160121
http://pythondatascience.plavox.info/scikit-learn/%E3%82%AF%E3%83%A9%E3%82%B9%E3%82%BF%E5%88%86%E6%9E%90-k-means

###補足情報(言語/FW/ツール等のバージョンなど)
OS X
python2系

Answer 1

簡単に実装するなら

colors = ['red','blue','yellow','pink','green']

のように色のリストを作成して

# 可視化（散布図）
for i in range(N_CLUSTERS):
    labels = features[pred == i]
    plt.scatter(labels[:, 0], labels[:, 1], c=colors[i])

のように散布図の描写毎に違う色を指定するとよいと思います

更に説明すると

from matplotlib.colors import ListedColormap
colors = ['red','blue','yellow','pink','green']
cmap = ListedColormap(colors)

のように独自のカラーマップ(数に対応する色のマップ)を作成しておくことで、散布図の描写を

# 可視化（散布図）
plt.scatter(features[:, 0], features[:, 1], c=pred, cmap=cmap)

のようにループを使わずに１行で c にクラスタ番号を指定するだけで書けるようになります。
これは重心の描写でも

# クラスタのセントロイド (重心) を描く
centers = cls.cluster_centers_
plt.scatter(centers[:, 0], centers[:, 1], s=80, marker='p', edgecolor='k', c=range(N_CLUSTERS), cmap=cmap)

のように使うことができるかと思います。

あと完全に補足ですが

cust_array = np.array([cust_df['Fresh'].tolist(),
                       cust_df['Milk'].tolist(),
                       cust_df['Grocery'].tolist(),
                       cust_df['Frozen'].tolist(),
                       cust_df['Detergents_Paper'].tolist(),
                       cust_df['Delicassen'].tolist()
                       ], np.int32)

の部分は

# DataFrameからNumpy配列を作成
cust_array  = cust_df.as_matrix().T

と書けますね

Comments

[SAMUSAMUN]

とてもわかりやすく教えていただきありがとうございます．
簡単に変更することができました．

Answer 2

plt.scatter(labels[:, 0], labels[:, 1],c=red)

scatter内にcの属性を入れてみても不可能ですか？

matplot(散布図)

Comments

[SAMUSAMUN]

ご指摘ありがとうございます．
色は変更できたのですが，クラスタ毎に色を変えたく思っています．
図で表した時に，色別になっているとわかりやすいからです
もっと自分でも調べてみようと思います