大熊貓數據框中使用列作爲字典的列表鍵

Question

我有以下數據框：

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

無論如何我可以做到這一點。我擁有的數據框是相當大的。

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