python pandas agg 和 count 在同一行

Question

我有 2 列数据，希望根据第 1 列对第 2 列进行分组，并使用其计数创建新的第 3 列。我做到了，

import pandas as pd
df = pd.read_csv("pairs.txt",sep='\t')
group = df.groupby(['COL_A'])['COL_B'].agg('-*-'.join)
group1 = df.groupby(['COL_A'])['COL_A'].count()

这样我就清楚地得到了组，

COL_A
A Cat-*-Rat-*-Mouse
B Monkey
C Tiger
D Squirrel

和组1:

COL_A
A 3
B 1
C 1
D 1

如何以更Pythonic的方式将group和group1组合成单行，以便我的输出是？

COL_A    COL_B         COUNT
    A Cat-*-Rat-*-Mouse 3
    B Monkey            1
    C Tiger             1
    D Squirrel          1

提前致谢。

Answer 1

试试这个:

group = df.groupby(['COL_A']).agg({'COL_B': ['-*-'.join, 'count']})

你是这个意思吗？