编程语言
首页 > 编程语言> > python – pandas plot dataframe barplot with category by category

python – pandas plot dataframe barplot with category by category

作者:互联网

我想用pandas绘制一个不同颜色的条形图,用于列中的类别.

这是一个简单的例子:(索引是可变的)

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

我想制作一个带有着色组的条形图.我还想指定颜色.在我的原始数据集中,我有很多goups.
有人可以帮我吗?

解决方法:

只需将颜色参数传递给绘图函数,并使用颜色列表:

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

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

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

而不是列表理解.

标签:python,pandas,plot,bar-chart
来源: https://codeday.me/bug/20191004/1852713.html