gpt4 book ai didi

python - 使用列作为键的 pandas DataFrame 到 dict 列表

转载 作者:太空宇宙 更新时间:2023-11-03 17:15:32 24 4
gpt4 key购买 nike

我有以下数据框:

ID     Col1    Col2
1 "A" "Z"
1 "A" "Y"
1 "B" "Z"
2 "A" "X"
2 "C" "P"

我想将上面的内容转换为字典形式的列表,如下所示,并按 ID 列计数:

[{"A" : 2, "B" : 1, "Z" : 2, "Y" : 1}, {"A" : 1, "C" : 1, "X" : 1, "P" : 1}]

无论如何我可以实现这一目标吗?我拥有的数据框相当大。

最佳答案

假设您的数据框名为df。您可以使用 df.groupby('ID').groups 获取每个 ID 的行号:

group_rows = df.groupby('ID').groups

我们将迭代每个组 ID,然后使用 CounterCol1Col2 中的值进行计数。然后我会将这些添加到字典中。

from collections import Counter

my_dict = {}
for group_id, rows in group_rows.iteritems():
c = Counter(df.iloc[rows, 1]) # 1 = index number for `Col1`
c.update(df.iloc[rows, 2]) # 2 = index number for `Col2`
my_dict[group_id] = dict(c)

>>> my_dict
{1: {'A': 2, 'B': 1, 'Y': 1, 'Z': 2},
2: {'A': 1, 'C': 1, 'P': 1, 'X': 1}}

我选择将结果输出到字典而不是您请求的列表,以便组 ID 和计数值之间的关系是明确的。如果这是一个问题,我可以假设数据帧首先按 ID 列排序,将它们转换为列表。

keys = my_dict.keys()
keys.sort()
my_list = [my_dict[k] for k in keys]
>>> my_list
[{'A': 2, 'B': 1, 'Y': 1, 'Z': 2}, {'A': 1, 'C': 1, 'P': 1, 'X': 1}]

关于python - 使用列作为键的 pandas DataFrame 到 dict 列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33696951/

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