matplotlibのグラフに、2つの単位を入れたいです。

Question

matplotlibで生成されるグラフに、2つの単位を入れたいです。具体的に言えば、

上のグラフの右側に、male, femaleそれぞれの青赤のグラフのパーセンテージを入れたいです。コードは、

sns.countplot('Sex', hue = 'Survived', data = train_data)

です。

パーセンテージを使ったグラフも生成しました。

get_legend_handles_labels()

を用いてみたのですが、できませんでした。

この二つの単位を、一つのグラフに統合したいです。その際、二つ目のグラフのmaleとfemaleの入れ替え方もご教示いただけると幸いです。

長くなってしまい、申し訳ございません。宜しくお願いします。

Answer 1

# data作成
import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np


"""
Sex,Survived
male,0
female,0
male,1
female,1
"""

Sex=['male']*480 +['male']*120+['female']*80+['female']*230
Survived=[0]*480+[1]*120+[0]*80+[1]*230
d={'Sex': Sex, 'Survived': Survived}

df=pd.DataFrame(data=d)

# いったんグラフの確認

sns.countplot(x='Sex', hue = 'Survived', data = df)

# パーセントのデータの作成
##　集計値
sub_df=df.value_counts().to_frame()
sub_df.columns=['Value']
##　小計
total_df=sub_df.groupby(level=0).sum()
total_df.columns=['Total']
##　結合
dfs = sub_df.join(total_df,on='Sex')

## インデックス及び値の順番の修正
dfs.sort_values(by=['Sex','Survived'],ascending=[False,True],inplace=True)
dfs.reset_index(inplace=True)

## パーセント値の計算
dfs['Percent']=round(dfs.Value / dfs.Total * 100,2)
## 必要なデータのみとする
dfs=dfs[['Sex','Survived','Percent']]
## グラフ用にデータを並び替える。
dfs.sort_values(by='Survived',inplace=True)

# 作図
fig=plt.figure(figsize=(10,10))
ax=sns.countplot(x='Sex', hue = 'Survived', data = df)

# 参照元
# https://stackoverflow.com/questions/33179122/seaborn-countplot-with-frequencies

for i,p in enumerate(ax.patches):
    x=p.get_bbox().get_points()[:,0]
    y=p.get_bbox().get_points()[1,1]
    ax.annotate('{:.1f}%'.format(dfs.iloc[i,].Percent), (x.mean(), y), 
            ha='center', va='bottom')

hueがSurvivedのため、パーセントデータは並び変えないとダメでした。
いかがでしょうか？

Comments

[rink]

ありがとうございます！！！全然大丈夫です！！！
ソウデスヨネ、hueで分けてる時点で、共通のパーセンテージの軸は作成できないですよね。。toshikawaさんのおかげで気づくことができました。ありがとうございます！
良ければ、https://teratail.com/questions/345510　もお分かりでしたら、ご教示願えないでしょうか。。宜しくお願いします！！