带注释的水平条形图答案

【问题标题】：Horizontal barplot with annotations带注释的水平条形图
【发布时间】：2017-08-09 05:15:58
【问题描述】：

当我每次运行这段代码时，显然 python 内核都死了并且死了，我成功地运行了代码：是代码有问题还是问题更深？我可以毫无问题地运行其他笔记本。

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)

【问题讨论】：

您是否收到任何错误信息？如果您使用任务管理器运行该进程，您的计算机处理该进程是否正常？
运行代码时遇到同样的问题，内核死了……奇怪。但是，在开始定义函数并在绘制之前分配垃圾变量时，一切运行良好。
@schlump 这非常奇怪，我能够运行这段代码，因为它现在在底部有垃圾。我记得我改变的最后一件事是 plt.barh() 行但不确定是什么（内核死后无法恢复）。
好吧..似乎是绘图的问题...在 plt.yticks() 之后添加 plt.show() 时，一切也运行良好...似乎内核并不是真的喜欢在情节仍然“开放”时运行代码
好吧，我照你说的做了，它不再死了，但是，哇，现在的结果太糟糕了。剧情前有巨大的空白，我做错了什么

标签： python pandas matplotlib kernel anaconda

【解决方案1】：

也许您应该更改问题的标题，因为对您而言，内核为何死亡并不重要。据我了解，问题是：

创建具有不同条形颜色和每个条形值的水平条形图作为注释。

这是使用 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')

创建：

更新：还可以为文本着色：

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

使图更大，以便背景也覆盖注释：

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

【讨论】：

对，这解决了问题，但是我想再问一件事，有没有办法像条形一样给注释着色？我尝试添加color=colors，但没那么简单。
耶哈颜色是一个字符串列表，你在循环中设置注释，你需要传递1个值。
什么意思？
查看更新，您需要将 1 个值传递给 color kwargs。因为您正在为循环中的每个注释设置颜色。所以颜色需要是“绿色”或“红色”...在我们创建颜色列表之前 ['green','red']
好吧，这行得通，顺便说一句，你知道需要提供什么类型的变量作为 clr 吗？因为我用一个具有三列的数据框重做这个条形图，现在我用与条形图本身不同的列来注释条形图，所以不能使用 p.width 颜色，但需要将该列转换为可用作此处选择颜色的类型。

【解决方案2】：

太棒了！我可以在水平条形图上写注释。那么，如果我有多个酒吧，我该怎么办？ seaborn horizontal barplot with annotation

经过多次尝试我的代码：

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

结果是：

【讨论】：