如何从按间隔分为两列的数据框创建分组条形图

时间:2019-06-30 04:22:50

标签: python pandas plot group-by

我将数据分为两列('a''b')。对于每一列,我将其切成垃圾箱。 对于每个交叉的垃圾箱,我希望有'c'的总和。然后,我想为'c'的总和创建一个分组图。但是,当我在分组数据之后使用.plot(kind='bar')时,这些条并没有分组在一起。

grouped = df.groupby([pd.cut(df['a'], [0, 1, 2, 3]), pd.cut(df['b'], [0, 101, 300, 500])])
grouped = grouped.c.sum()
grouped.plot(kind='bar')

ungrouped barplot

我希望将图表按'a'分组,并用不同的颜色代表'b'。我已经尝试过.unstack().plot(kind='bar');但是,它返回错误:

  

无法确定类型的下一个标签

1 个答案:

答案 0 :(得分:0)

给出以下示例数据

df = pd.DataFrame({'a': [8, 49, 81, 52, 22, 24, 32, 67, 24, 21, 78, 16, 86, 19, 4, 88, 4, 20, 20, 1],
                   'b': [2, 49, 49, 70, 31, 47, 98, 26, 55, 36, 17, 39, 56, 80, 52, 36, 42, 69, 73, 70],
                   'c': [22, 99, 31, 95, 16, 32, 81, 20, 58, 23, 53, 5, 0, 81, 8, 68, 16, 36, 35, 54]})

和您的设置

grouped = df.groupby([pd.cut(df['a'], [16, 30, 67, 86]),
                      pd.cut(df['b'], [0, 26, 49, 100])])
grouped = grouped['c'].sum()

您有两个选择

  • (grouped.reset_index()
            .pivot('a', 'b')
            .plot(kind='bar'));
    

    enter image description here

  • grouped = grouped.unstack()
    grouped.columns = grouped.columns.categories
    grouped.plot(kind='bar');
    

    enter image description here