我想了解如何在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)
目前没有回答
相关问题 更多 >
编程相关推荐