代码之家  ›  专栏  ›  技术社区  ›  Michael

使用pandas按列值分组条形图

  •  1
  • Michael  · 技术社区  · 6 年前

    我正试图为编码为列值的类别显示分组条形图( col2 在我的例子中)

    条形图的图例必须显示编码为 二氧化碳 所以 True False 应该根据这些值对条进行分组。

    df = pd.DataFrame({
        'col1': ['A', 'B', 'A', 'C', 'B', 'C', 'A', 'A', 'A', 'A', 'B'],
        'col2': [True, False, True, True, False, True, False, False, False, False, True],
    })
    df.groupby('col1')['col2'].count().plot(kind = 'bar', legend=True)
    

    enter image description here

    相反,我想看到的是:

    enter image description here

    1 回复  |  直到 6 年前
        1
  •  1
  •   Joe    6 年前

    我认为你需要:

    df.groupby(['col1', 'col2']).agg('size').unstack().plot(kind = 'bar', legend=True)
    

    enter image description here