【Python】回帰モデルの描画方法｜seaborn基礎

seabornを使用すると、Pythonを使用してデータを可視化することができます。

本記事では、seabornによる回帰モデルの描画方法について、詳しくご説明します。

levtech-ad

＼業界最大級の案件数！／

レバテックフリーランス詳細はコチラ＞＞

seabornとは

seabornは、Pythonを使用してデータを可視化するための外部ライブラリの１つです。

データ可視化用ライブラリは他にも複数あり、その中でも特にMatplotlibが有名です。

seabornはMatplotlibをベースに作られており、Matplotlibの描画機能を利用しています。

seabornの長所は、Matplotlibよりも美しい図を、より少ないコードで簡単に描ける点です。

本記事では、seabornによる回帰モデルの描画方法をご紹介します。

seabornのインストール

「seaborn」は、以下コマンドを入力することで、インストールすることができます。

コマンドの入力は、コマンドプロンプトあるいはターミナルから行います。

pip install seaborn

動作確認として、試しに以下を入力します。

import seaborn as sns

上記を入力してもエラーが発生しなければ、正常にインストールされています。

回帰モデルの描画方法

seaborn.lmplot()関数を使用することで、2次元データと回帰モデルを同時に描画することができます。

またFacetGrid機能により、複数グラフをまとめて出力することができます。

lmplot()関数の引数には、x軸（横軸）の変数、y軸（縦軸）の変数、データセット等を指定します。

データセットの「penguins」を用いて、回帰モデルを描画してみます。

#input
import seaborn as sns
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

# データセットの読み込み
penguins = sns.load_dataset("penguins")

# lmplotの指定・描画
sns.lmplot(data=penguins, x="bill_length_mm", y="bill_depth_mm")
plt.show()

hueを指定すると、カテゴリ別に描画できます。

#input
import seaborn as sns
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

# データセットの読み込み
penguins = sns.load_dataset("penguins")

# lmplotの指定・描画
sns.lmplot(data=penguins, x="bill_length_mm", y="bill_depth_mm", hue="species")
plt.show()

colを指定すると、列方向に複数のサブプロットを持つ複数グラフを描画できます。

#input
import seaborn as sns
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

# データセットの読み込み
penguins = sns.load_dataset("penguins")

# lmplotの指定・描画
sns.lmplot(data=penguins, x="bill_length_mm", y="bill_depth_mm", 
           hue="species", col="sex", height=4)
plt.show()

rowを指定すると、行方向に複数のサブプロットを持つ複数グラフを描画できます。

#input
import seaborn as sns
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

# データセットの読み込み
penguins = sns.load_dataset("penguins")

# lmplotの指定・描画
sns.lmplot(data=penguins, x="bill_length_mm", y="bill_depth_mm", 
           col="species", row="sex", height=3)
plt.show()

サブプロット間の軸の制限を指定することができます。

#input
import seaborn as sns
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

# データセットの読み込み
penguins = sns.load_dataset("penguins")

# lmplotの指定・描画
sns.lmplot(data=penguins, x="bill_length_mm", y="bill_depth_mm", 
           col="species", row="sex", height=3, 
           facet_kws=dict(sharex=False, sharey=False))
plt.show()

まとめ

この記事では、seabornによる回帰モデルの描画方法について、ご説明しました。

本記事を参考に、ぜひ試してみて下さい。

参考

Python学習用おすすめ教材

Pythonの基本を学びたい方向け

統計学基礎を学びたい方向け

Pythonの統計解析を学びたい方向け

おすすめプログラミングスクール

テックアカデミーの無料体験