Groupby dataframe生成包含不同组和计数的单个列表列

你可以试试群比 id 列,然后将 grp 和 val 柱

out = (have.groupby('id')
       .apply(lambda g: list(zip(g['grp'], g['val'])))
       .rename('result')
       .reset_index())

print(out)

   id                    result
0   1  [(a, 5), (b, 4), (c, 3)]
1   2          [(d, 2), (e, 1)]

如果要将两个以上的列压缩到元组列表中,还可以使用 df.itertuples 但是 df.to_records 在别人的回答中引用也可以。

out = (have.groupby('id')
       .apply(lambda g: list(g[['grp', 'val']].itertuples(index=False)))
       .rename('result')
       .reset_index())

打印(输出)

01[(a,5)、(b,4)、(c,3)]
12[(d,2)、(e,1)]

将数据作为列表元组的一种方法是使用 df.to_records 然后 groupby.agg .

have.assign(
    res=have[["grp", "val"]].to_records(index=False).tolist()
).groupby("id", as_index=False)["res"].agg(list)


#    id                       res
# 0   1  [(a, 5), (b, 4), (c, 3)]
# 1   2          [(d, 2), (e, 1)]

你可以使用:

want = (have
 .assign(result=have[['grp','val']].agg(tuple, 1))
 .groupby('id')['result']
 .agg(list).reset_index()
 )