matplotlibでデータを可視化する

AI開発やデータサイエンスにおいてデータの可視化は欠かすことができません。数値で表現された無数のデータも散布図などにプロットすることにより、データ全体が持つ特徴を人間の目でも確認することができるようになります。

gnuplotなど有名なプロットツールもありますが、pythonとmatplotlibを使えば簡単にグラフやプロット図を描画することができます。今回は、matplotlibを使って３次元散布図を作成していきます。

1 matplotlibでデータを可視化する
2 グラフを画面上に表示させる

matplotlibでデータを可視化する

アヤメのデータを使用して３次元散布図を作成してみます。アヤメのデータの作成については、下のダウンロードボタンから取得できます。

ダウンロード

散布図を作成するpythonコードは以下の通りです。

import pandas as pd
from sklearn.model_selection import train_test_split
import pydotplus as pdp
from sklearn.externals.six import StringIO
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

csv = pd.read_csv('iris.csv')

csv_data = csv[["SepalLength","SepalWidth","PetalLength","PetalWidth"]]
csv_label = csv["Name"]

train_data, test_data, train_label, test_label = \
train_test_split(csv_data, csv_label, test_size=0.3, random_state=1)

labels = train_label.map({'Iris-setosa': 0, 'Iris-versicolor': 1, 'Iris-virginica': 2})
aaa = train_data[["PetalWidth","PetalLength","SepalLength"]]

fig = plt.figure()
ax = Axes3D(fig)
ax.scatter(aaa[["PetalWidth"]],aaa[["PetalLength"]],aaa[["SepalLength"]],c=[['orange', 'green', 'blue'][x] for x in labels])

ax.set_title("Iris dataset 3D scatter plot")
ax.set_xlabel("PetalWidth")
ax.set_ylabel("PetalLength")
ax.set_zlabel("SepalLength")

plt.show()

スクリプトの前半でアヤメのデータを読み込んでいます。また、「がくの長さ」、「がくの幅」、「花びらの長さ」と「花びらの幅」の中から、特徴量の重み順にデータを取り出しています。以下がその部分です。

aaa = train_data[["PetalWidth","PetalLength","SepalLength"]]

matplotlibのグラフ図設定は、以下の通りです。

import pydotplus as pdp
from sklearn.externals.six import StringIO
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

・・・

fig = plt.figure()
ax = Axes3D(fig)
ax.scatter(aaa[["PetalWidth"]],aaa[["PetalLength"]],aaa[["SepalLength"]],c=[['orange', 'green', 'blue'][x] for x in labels])

ax.set_title("Iris dataset 3D scatter plot")
ax.set_xlabel("PetalWidth")
ax.set_ylabel("PetalLength")
ax.set_zlabel("SepalLength")

plt.show()

scatter()は散布図作成のメソッドで、特徴量の重み順に抽出した要素をx、y、zの軸に割り当て、アヤメの種類ごとにオレンジ色、緑色及び青色でプロットします。また、set_titleはグラフタイトル、set_xlabel、set_ylabelやset_zlabelで各軸のタイトルを設定します。

では、実際にどのような図が表示されるのか見てみましょう。