pandas DataFrame以列为键列出dict

Question

我有以下数据帧：

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

我想将dict表单列表中的上述内容转换为以下ID列计数：

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

无论如何，我能做到这一点吗。我的数据帧相当大。

Answer 1

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

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

我们将遍历每个组ID，然后使用Counter来计算Col1和Col2中的值。然后我把这些加到字典里。

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}]