pandas按类别绘制带颜色的数据框条形图

2024-03-29 13:41:28 发布

您现在位置:Python中文网/ 问答频道 /正文

我想用熊猫来绘制一个柱状图,柱状图中的类别有不同的颜色。

下面是一个简单的例子:(index是variable)

df:
         value   group
variable               
a             10      1
b              9      1
c              8      1
d              7      2
f              6      2
g              5      3
h              4      3

我想做一个组着色的条形图。我还想指定颜色。在我的原始数据集中,我有很多痛风。 有人能帮我一下吗?


Tags: df原始数据indexvalue颜色group绘制类别
1条回答
网友
1楼 · 发布于 2024-03-29 13:41:28

只需向plot函数传递一个颜色参数和一个颜色列表:

df['group'].plot(kind='bar', color=['r', 'g', 'b', 'r', 'g', 'b', 'r'])

如果要将value绘制为条形图,并且还希望group确定条形图的颜色,请使用:

colors = {1: 'r', 2: 'b', 3: 'g'}
df['value'].plot(kind='bar', color=[colors[i] for i in df['group']])

您还可以使用类似于:

list(df['group'].map(colors))

而不是列表理解。

相关问题 更多 >