python - 如何使用 group 对分类列 pandas 进行分组

Question

我有一个数据框 A，它看起来像这样：

col1    col2
 1       low
 1       low
 1       high
 2       low
 2       high

现在，我想将这个数据框分组，同时在单独的列col1中获取每个元素的计数。col2因此，生成的数据框 B 应如下所示：

 col1     low    high
  1        2       1
  2        1       1

我正在尝试通过使用来做到这一点group_by，但这不起作用。任何人都知道如何做到这一点？

Answer 1

这是我尝试使用value_counts-> reset_index-> pivot-> apply( dropna) ->astype(int)

df2.value_counts().reset_index().pivot(columns='col2',values=0).apply(lambda x: pd.Series(x.dropna().values)).astype(int)