python - 条形图 : How to choose color if value is positive vs value is negative

Question

我有一个包含正值和负值的 pandas 数据框，我想将其绘制为条形图。

我想绘制正颜色“绿色”和负值“红色”(非常原始......哈哈)。

我不确定如何通过 if > 0 'green' else < 0 'red'？

data = pd.DataFrame([[-15], [10], [8], [-4.5]],
                    index=['a', 'b', 'c', 'd'],
                    columns=['values'])
data.plot(kind='barh')

Answer 1

我会为观察值是否大于 0 创建一个虚拟列。

In [39]: data['positive'] = data['values'] > 0

In [40]: data
Out[40]: 
   values positive
a   -15.0    False
b    10.0     True
c     8.0     True
d    -4.5    False

[4 rows x 2 columns]

In [41]: data['values'].plot(kind='barh',
                             color=data.positive.map({True: 'g', False: 'r'}))

此外，您可能需要注意不要让列名与 DataFrame 属性重叠。 DataFrame.values为 DataFrame 提供底层的 numpy 数组。名称重叠会阻止您使用 df.<column name>语法。