python 特定のカラム値に限定して他カラムの要素と数量を可视化する

Posted 2021-05-09

# numpy, pandas, matplotlib.pyplot, seaborn必要

# あるカラムから特定の要素を抽出する
cause = df[
    df['Type_code'] == 'Causes'
]

# 特定した要素の行で、別カラムを参照し、そこの要素と数量を特定・バーグラフで可視化する
def plot_type(df, Title, x_lab):
    p_type = df.groupby('Type').sum()['Total']
    sort_df = p_type.sort_values(ascending = False)
    
    fig = sort_df.plot('bar', figsize = (10,6), title = Title, width = 0.75)
    fig.set_xlabel(x_lab)
    fig.set_ylabel('Count')
    sns.set_style('whitegrid')
    sns.set_palette('Set2')

# セットした要素を可視化
plot_type(cause, 'Suicide by cause', 'Cause')