実データの解析(その3)

多変量解析・機械学習の実装に便利な言語・ソフトやサンプルスクリプト、解析例を紹介します。

スクリプト(2) Pythonによる相関係数の計算

フィッシャーのアヤメのデータ(萼片の長さ、萼片の幅、花弁の長さ、花弁の幅の計4項目)

※Python組み込みのデータを読み込み、相関係数を計算し、ヒートマップを描く。

import pandas as pd #データ加工用のライブラリの読み込み
import seaborn as sns #データ可視化用のライブラリの読み込み
import matplotlib.pyplot as plt #グラフ描画用ライブリの読み込み
from sklearn import datasets #機械学習用のライブラリからデータセットの読み込み

iris = datasets.load_iris() #アヤメのデータの読み込み
df = pd.DataFrame(iris.data, columns=iris.feature_names) #データの型変換
corr= df.corr() #相関係数を計算
sns.heatmap(corr, square=True, annot=True) #ヒートマップを描画
plt.show() #グラフ表示

ヒートマップ

 

次のページに続く

▲このページのトップへ