基于csv文件的Python字典更新

Question

我有一个 csv 文件，其中:

id, word, super_id
100 , cat, 5
5 , bird, Nan
3 , dog, 100
20, fox , 100

我创建了以下字典:d= {100: [5, 3, 20], 3: [100]}

• 其中节点 100 有父节点 5 和子节点 3、20
• 节点 3 有父节点 100，没有子节点

我需要更新字典，使其变成:d= {猫:[鸟，狗，狐狸]，狗:[猫]}或创建一个新的。有什么想法吗？

Answer 1

这是使用 csv 模块和一些字典理解的一种方式:

import csv
from io import StringIO

mystr = StringIO("""id, word, super_id
1 , cat, 2
2 , bird, Nan
3 , dog, 1
4, fox , 1""")

# replace mystr with open('file.csv', 'r')
with mystr as fin:
    reader = csv.reader(fin, skipinitialspace=True)
    next(reader)  # ignore header row, or use headers = next(reader) to extract
    fin_d = {int(ide): word for ide, word, super_id in reader}

d = {1: [2, 3, 4], 3: [1]}

# map keys and values of d using fin_d
res = {fin_d[k]: list(map(fin_d.get, v)) for k, v in d.items()}

print(res)

{'cat': ['bird', 'dog', 'fox '], 'dog': ['cat']}