使用 plotnine 时基于 pandas 数据框中的值的颜色条

Question

我正在尝试使用 plotnine 构建瀑布图。我想将开始和结束条着色为灰色（理想情况下我想指定十六进制颜色），增加为绿色，减少为红色。

下面是一些示例数据和我当前的情节。我正在尝试将 fill 设置为 pandas 列 colour，但这些条都是黑色的。我也将 fill 放在 geom_segment 中，但这也不起作用。

df = pd.DataFrame({})
df['label'] = ('A','B','C','D','E')
df['percentile'] = (10)*5
df['value'] = (100,80,90,110,110)
df['yStart'] = (0,100,80,90,0)
df['barLabel'] = ('100','-20','+10','+20','110')
df['labelPosition'] = ('105','75','95','115','115')
df['colour'] = ('grey','red','green','green','grey')

p = (ggplot(df, aes(x=np.arange(0,5,1), xend=np.arange(0,5,1), y='yStart',yend='value',fill='colour'))
    + theme_light(6)
    + geom_segment(size=10)
    + ylab('value')
    + scale_y_continuous(breaks=np.arange(0,141,20), limits=[0,140], expand=(0,0))
)

编辑

根据 teunbrand 将 fill 更改为 color 的评论，我有以下内容。如何指定实际颜色，最好是十六进制格式？

Answer 1

为了结束这个问题，在解决方案的评论中归功于 teunbrand。

geom_segment() 有 colour 美感但没有 fill 美感。将 fill='colour' 替换为 colour='colour'.

Plotnine 将为条形图使用默认颜色。如果 DataFrame 列的内容是文字颜色，则使用 scale_color_identity()，或者使用 scale_colour_manual() 手动指定一个元组或颜色列表。两种形式都接受十六进制颜色。

使用 plotnine 时基于 pandas 数据框中的值的颜色条

Colour bars based on values in pandas dataframe when using plotnine

python

r

ggplot2

python-3.x

plotnine