问题描述
给定一个pandas dataframe ，其外观如下column_a和column_b 。
我该如何构造另外2列，一列为所有列计算column_a中每个值的频率，另一列为column_a中的值相同的地方计算值的唯一数目：
column_a | column_b | col_a_count | count_unique_b_where_a
0 1 4 3
0 1 4 3
0 2 4 3
0 3 4 3
2 0 3 1
2 0 3 1
2 0 3 1
5 3 1 1
9 5 6 5
9 5 6 5
9 3 6 5
9 4 6 5
9 2 6 5
9 1 6 5
1楼
使用groupby和agg ：
s = (df.groupby('column_a').agg(
{'column_a': 'count', 'column_b': 'nunique'}).reindex(df.column_a))
column_a column_b
column_a
0 4 3
0 4 3
0 4 3
0 4 3
2 3 1
2 3 1
2 3 1
5 1 1
9 6 5
9 6 5
9 6 5
9 6 5
9 6 5
9 6 5