python - 逻辑矩阵的 Pandas Groupby 条形图

Question

a 0 0 0 0 1  
a 0 0 0 1 1  
a 1 0 0 1 1  
b 1 0 0 1 1  
c 0 0 1 1 1  
c 1 0 0 0 0   
c 0 1 0 0 0  

如果我有一个像上面那样的 Pandas 数据框，我该如何按 a、b 和 c 对数据进行分组，然后绘制每列的总和/计数？

理想情况下，我的条形图中有 3 个组，每个组有 5 个条形代表该组中该列的总和/计数。

Answer 1

您想对数据框使用 groupby 方法来创建数据框 groupby 对象。在 groupby 对象上调用 sum 会创建一个新的数据框，它是原始数据框的枢轴。从那里您可以调用 plot 来制作条形图。

import pandas as pd

d = {'type': ['a','a','a','b','c','c','c'],
     'v1': [0,0,1,1,0,1,0],
     'v2': [0,0,0,0,0,0,1],
     'v3': [0,0,0,0,1,0,0],
     'v4': [0,1,1,1,1,0,0],
     'v5': [1,1,1,1,1,0,0]}

df = pd.DataFrame(d)
gb_sum = df.groupby('type').sum()
gb_sum.plot(kind='bar')