python3:cs中多个键的嵌套字典

Question

数据如下：

id,outer,inner1,inner2,inner3
123,"Smith,John",a,b,c
123,"Smith,John",d,e,f
123,"Smith,John",g,h,i
456,"Williams,Tim",xx,yy,zz
456,"Williams,Tim",vv,ww,uu
456,"Miller,Ray",rrr,sss,ttt
456,"Miller,Ray",qqq,www,ppp

我希望生成的词典

^{pr2}$

我尝试调整来自Python Creating A Nested Dictionary From CSV File的已接受答案，但此方法在每一行都覆盖字典，因此只有每个id的最后一行才会出现在字典中。在

Answer 1

一个collections.defaultdict使用每行的第一个元素作为外部dict键，然后使用第二个元素作为内部dict键，并将行中的其余值添加到列表中作为内部dict键的值：

import csv
from collections import defaultdict
with open("in.txt" ) as f:
    next(f) # skip header
    d = defaultdict(lambda: defaultdict(list))
    r = csv.reader(f)
    for row in r:
        d[row[0]][row[1]].extend(row[2:])

from pprint import pprint as pp

pp(dict(d))

输出：

^{pr2}$

因为您使用的是python3，所以我们可以在循环中使用*进行解包，使代码更漂亮：

with open("in.txt") as f:
    next(f)  # skip header
    d = defaultdict(lambda: defaultdict(list))
    r = csv.reader(f)
    for k1, k2, *vals in r:
        d[k1][k2].extend(vals))