向Seaborn条形图添加文本

时间:2018-06-24 00:54:27

标签: python plot bar-chart seaborn

我希望了解如何向海洋图添加文字。请参阅下面的我的功能。我希望将“ coef”列添加到条形图的内部。我在网上四处张望,努力了解如何将其整合。就matplotlib的工作方式而言,各种“ ax”和子图都使我感到困惑。我试图使用“ text”功能,但似乎无法正确处理。

import seaborn as sns
import pandas as pd
import numpy as np

df= pd.DataFrame({'variable':['a','b','c','d'], 'coef':[2,-2,3.5, -.09]})

def myplot_coef(df, variables_col, coef_col, diverging=False):

    df['abs_coef']=np.abs(df[coef_col])

    df.sort_values(by='abs_coef', ascending=False,inplace=True)
    df.reset_index(inplace=True, drop = True)

    plotcolors = ['grey' if x < 0 else 'green' for x in df[coef_col]]


    if diverging == True:
        g = sns.barplot(x=coef_col,y=variables_col, data=df, palette=plotcolors )
    elif diverging == False:
        g = sns.barplot(x='abs_coef',y=variables_col, data=df, palette=plotcolors )

    return(df)

myplot_coef(df=df, variables_col='variable', coef_col='coef', diverging=True)

0 个答案:

没有答案