我有一个包含两列的 CSV,第一列是致力于我们项目中特定建筑的团队。
第二列是实际的建筑物编号。
我正在寻找的是一本字典,其中第一列作为键,列表中属于该团队的建筑物。
我尝试了各种形式的csv.reader和csv.DictReader以及不同的 for 循环将数据重写到另一个字典,但我无法获得我想要的结构。
CSV:
team,bldg,
3,204,
3,250,
3,1437,
2,1440,
1,1450,
Run Code Online (Sandbox Code Playgroud)
字典的结构如下:
dict["1"] = ["1450"]
dict["2"] = ["1440"]
dict["3"] = ["204", "250", "1437"]
Run Code Online (Sandbox Code Playgroud)
这有效:
import csv
result={}
with open('/tmp/test.csv','r') as f:
red=csv.DictReader(f)
for d in red:
result.setdefault(d['team'],[]).append(d['bldg'])
#results={'1': ['1450'], '3': ['204', '250', '1437'], '2': ['1440']}
Run Code Online (Sandbox Code Playgroud)