Python_(1/4)エクセルで書けないグラフを作成(seaborn/stripplot) | バイオとDX

バイオとDX

バイオ医薬品とDX、データサイエンス

# =============================================================================
# Seabornで、Stripplot
# =============================================================================
#タイトル行を1行目にしたcsvファイル(UTF-8)を作成。作業フォルダに保存しておく。

 

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

# CSVファイルを読み込み(タイトル行を下記で準備したcsv)
df_1 = pd.read_csv("data1.csv")

# stripplotを作成
ax = sns.stripplot(x="class", y="titer", hue="keiretsu", data=df_1)

#メジアン横線の幅
median_width = 0.2

for tick, text in zip(ax.get_xticks(), ax.get_xticklabels()):
    sample_name = text.get_text()  # "X" or "Y"

    # メジアンを計算 classごとにtiterのメジアンを算定
    median_val = df_1[df_1['class']==sample_name].titer.median()

    # メジアンの横線 lwは、lineidth
    ax.plot([tick-median_width/2, tick+median_width/2], [median_val, median_val],lw=2, color='k')

#plt.show()

# stripplotをPNG形式で保管
plt.savefig("data2.png")