带注释的水平条形图

Question

I was successfully running the code when apparently python kernel died and dies everytime I am running this code: Is it something wrong with the code or the problem is deeper? I can run other notebooks without problems.

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
%matplotlib inline
plt.rcParams['text.usetex'] = False



df = pd.DataFrame(np.random.uniform(size=37)*100, columns=['A'])

ax = plt.figure(figsize=(10,5))

plt.barh(df.index, df['A'], color='ForestGreen')
plt.yticks(df.index)


def annotateBars(row, ax=ax):
    if row['A'] < 20:
        color = 'black'
        horalign = 'right'
        horpad = 2
    else:
        color = 'white'
        horalign = 'right'
        horpad = -2

    ax.text(row.name, row['A'] + horpad, "{:.1f}%".format(row['A']),
         color=color,
            horizontalalignment=horalign,
            verticalalignment='center',
            fontsize=10)

junk = df.apply(annotateBars, ax=ax, axis=1)

Answer 1

Maybe you should change the title of the question, because as for you it does not really matter why the Kernel dies. As far as I've understood you, the problem is:

Creating a horizontal bar chart with different bar colors and the values of each bar as an annotation.

Here is a solution using Seaborn:

import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np 
import pandas 

sns.set(style="darkgrid")
sns.set_color_codes("muted")

# Create example DataFrame
df = pandas.DataFrame(np.random.uniform(size=20)*100, columns=['A']) 

# Create list of colors based on a condition 
colors = ['red' if (x < 20) else 'green' for x in df['A']]

# Create barplot 
ax = sns.barplot(data=df.transpose(), palette=colors, orient='h')
# Annotate every single Bar with its value, based on it's width           
for p in ax.patches:
    width = p.get_width()
    plt.text(5+p.get_width(), p.get_y()+0.55*p.get_height(),
             '{:1.2f}'.format(width),
             ha='center', va='center')

Creates:

Update: For also coloring the text:

for p in ax.patches:
    width = p.get_width()
    if width < 20:
        clr = 'red'
    else:
        clr = 'green'
    plt.text(5+p.get_width(), p.get_y()+0.55*p.get_height(),
             '{:1.2f}'.format(width),color=clr,
             ha='center', va='center')

Making the plot larger so the background also covers the annotations:

ax.set_xlim([0, max(df['A'])+10])

Answer 2

Great! I can write annotations on horizontal barplot. Well, what can i do if i have more than one bars? seaborn horizontal barplot with annotation

After many tries that worked on my code:

cnt=0
for p in ax.patches:
    width=p.get_width()
    if cnt%2==0:
        a=p.get_width()-5
        clr = 'purple'
    else:
        a=p.get_width()+5
        clr = 'blue'
    plt.text(a, 
    p.get_y()+0.55*p.get_height(),'{:1.2f}'.format(width),color=clr,ha='center', va='center')
    cnt=cnt+1

result is: