seabornでグラフを複数のグラフを描画する

概要

seabornで複数のグラフを複数行複数列で描画するときの設定や
y軸を揃える設定、グラフのサイズを変える設定の方法を記載します。

どうして書いたのか

seabornでグラフを複数描画するときに、
いつも「あれ、どうだったっけ？」ってなって検索するのに時間がかかるので、
他にも同じ人がいるかもと思い、記事にしました。

前提

• 実行環境は jupyter notebook です。
• import するのは以下。

import matplotlib.pyplot as plt
import seaborn as sns

• 描画するデータは以下で取得できるseabornのサンプルデータを使います。

tips_df = sns.load_dataset('tips')
tips_df.head()

	total_bill	tip	sex	smoker	day	time	size
0	16.99	1.01	Female	No	Sun	Dinner	2
1	10.34	1.66	Male	No	Sun	Dinner	3
2	21.01	3.50	Male	No	Sun	Dinner	3
3	23.68	3.31	Male	No	Sun	Dinner	2
4	24.59	3.61	Female	No	Sun	Dinner	4

1行2列で描画する

subplotsに行, 列 な感じで数字を指定する。
(もし2行3列なら、 plt.subplots(2,3) となる。)

ax1とax2にsubplotsで指定した設定が入るイメージをぼくは持っています。
それを各グラフのaｘに指定することで設定を反映させます。
axがパラメータとしてないグラフもあります。lmplotとか。

fig, (ax1, ax2) = plt.subplots(1, 2)

sns.regplot(x='total_bill', y='size', data=tips_df, ax=ax1)
sns.countplot(x='size', data=tips_df, ax=ax2)

2つのグラフ間でy軸を揃える

subplotsのところに sharey=True を追加する。

fig, (ax1, ax2) = plt.subplots(1, 2, sharey=True)

sns.regplot(x='total_bill', y='size', data=tips_df, ax=ax1)
sns.countplot(x='size', data=tips_df, ax=ax2)

グラフをサイズを変える

subplotsのところに figsize=(10, 20) を追加する。
()の中の数字は(横のサイズ, 縦のサイズ) です。

fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(10,20))

sns.regplot(x='total_bill', y='size', data=tips_df, ax=ax1)
sns.countplot(x='size', data=tips_df, ax=ax2)