向seaborn barch添加文本

2024-09-25 10:23:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我想了解如何在seaborn情节中添加文本。请参阅下面的功能。我想把“coef”列添加到条形图中,就在条形图内部。我在网上到处看看,很难理解如何将其融入其中。在matplotlib的工作方式方面,各种各样的'ax'和subplot的东西让我感到困惑。。。我尝试过使用“文本”功能,但似乎无法正确使用。在

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)

Tags: 文本import功能falsetruedfascol