Добавить текст на морскую диаграмму - PullRequest
Новая
       21

Добавить текст на морскую диаграмму

0 голосов
/ 24 июня 2018

Я пытаюсь понять, как добавить текст к морским заговорам.Смотрите мою функцию ниже.Я собираюсь добавить колонку 'coef' к диаграммам прямо внутри бара.Я посмотрел вокруг в Интернете и изо всех сил пытаюсь понять, как включить это.Различные вещи типа «топор» и «подзаговор» смущают меня с точки зрения того, как работает matplotlib ... Я пытался использовать функцию «текст», но, похоже, не могу понять, как это правильно. 

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)
...