始める無料で始める

matplotlib でクラスターを可視化する

クラスターの妥当性を評価したり、データの傾向を見つけたりするには、可視化が欠かせないことを学びました。ここでは、matplotlib モジュールを使って、Comic-Con の来場者データ（footfall データセット）を可視化してみましょう。

データは pandas の DataFrame comic_con に保存されています。x_scaled と y_scaled は、ある時点における人の標準化された X・Y 座標の列名です。cluster_labels にはクラスターラベルが入っています。リンケージオブジェクトは変数 distance_matrix に保存されています。

この演習はコースの一部です

Pythonで学ぶクラスタ分析

コースを見る

演習の手順

  • matplotlib モジュールから pyplot クラスを plt としてインポートします。
  • 2 つのクラスターラベル 1 と 2 に対する colors 辞書を定義します。
  • colors 辞書で定義した各クラスターの色を使って散布図を描画します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Import the pyplot class
____

# Define a colors dictionary for clusters
colors = {____:'red', ____:'blue'}

# Plot a scatter plot
comic_con.plot.scatter(x=____, 
                	   y=____,
                	   c=comic_con['cluster_labels'].apply(____))
plt.show()
コードを編集して実行