python - Pandas 分组并记录分组值

Question

我有以下数据集:

Code    Value
100004  1
1017    1
1017    3
1071    1
1071    3
3039    1
3397    1
3397    3

我能够使用 pandasDataFrame.groupby('Code', as_index=False).agg('count') 计算值

我想计算并同时记录新列中的值。我希望结果看起来像下面的示例

Code       NoValues      Values
100004     1             1
1017       2             1,3
1071       2             1,3
3039       1             1
3397       2             1,3

可以用 pandas 做吗？

Answer 1

你可以使用groupby + agg:

df.Value.groupby(df.Code).agg({'Values': lambda g: list(g), 'NumValues': lambda g: len(g)}).reset_index()

例如:

df = pd.DataFrame({'Code': [1004, 1004, 1007], 'Value': [1, 2, 8]})
>>> df.Value.groupby(df.Code).agg({'Values': lambda g: list(g), 'NumValues': lambda g: len(g)}).reset_index()
    Code    NumValues   Values
0   1004    2   [1, 2]
1   1007    1   [3]