Python バイオリン図と箱ひげ図を同時描画する「ptitprince」

　本記事では、下図のようにviolin plot と box plot を同時プロットする雛形コードを載せました。ライブラリ「ptitprince」を使用します。

使用したサンプルデータは、機械学習でお馴染みのアイリスデータセットです(下図)。x軸にspecies列をy軸にsepal_width列を使用しました。

インストールはpipで次のようにします。

pip install ptitprince

■本プログラム

import ptitprince as pt
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
plt.rcParams['font.size']=16
sns.set(style='ticks', color_codes=True)
sns.set_context('talk')

df = pd.read_csv('iris-dataset.csv')
df

plt.title("iris-dataset")
pt.RainCloud(x="species", y="sepal_width", data=df, pointplot=True, point_linestyles='--')
ax = plt.gca()
y_min , y_max = ax.get_ylim()
ax.set_ylim(0, y_max)
plt.grid()
plt.tight_layout()

以上

<広告>

リンク