gpt4 book ai didi

python - 转换pandas.groupby为dict

转载 作者:行者123 更新时间:2023-12-04 16:57:40 30 4
gpt4 key购买 nike

考虑一下,数据帧d

d = pd.DataFrame({'a': [0, 2, 1, 1, 1, 1, 1],
'b': [2, 1, 0, 1, 0, 0, 2],
'c': [1, 0, 2, 1, 0, 2, 2]}
> a b c
0 0 2 1
1 2 1 0
2 1 0 2
3 1 1 1
4 1 0 0
5 1 0 2
6 1 2 2


我想按列 a将其拆分成这样的字典:

{0:    a  b  c
0 0 2 1,

1: a b c
2 1 0 2
3 1 1 1
4 1 0 0
5 1 0 2
6 1 2 2,

2: a b c
1 2 1 0}


我使用 pandas.groupby找到的解决方案是:

{k: table for k, table in d.groupby("a")}


还有哪些其他解决方案?

最佳答案

您可以将dicttuple / list应用于groupby一起使用:

res = dict(tuple(d.groupby('a')))

关于python - 转换pandas.groupby为dict,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50966466/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com