如何在 python3 中将有组织的文件制作成字典？

Question

我正在尝试制作这个文件：

c;f
b;d
a;c
c;e
d;g
a;b
e;d
f;g
f;d

变成这样的字典：

{'e': {'d'}, 'a': {'b', 'c'}, 'd': {'g'}, 'b': {'d'}, 'c': {'f', 'e'}, 'f': {'g', 'd'}}.

我现在使用的代码如下：

def read_file(file : open) -> {str:{str}}:
f = file.read().rstrip('\n').split()
answer = {}
for line in f:
    k, v = line.split(';')
    answer[k] = v
return answer

但它给了我{'f': 'g', 'a': 'c', 'b': 'd', 'e': 'd', 'c': 'e', 'd': 'g'}

我该如何解决？

Answer 1

字典覆盖前一个键，使用defaultdict here

>>> import collections 
>>> answer = collections.defaultdict(set)
>>> for line in f: 
...     k, v = line.split(";")
...     answer[k].add(v)
... 
>>> answer
defaultdict(<class 'set'>, {'b': {'d'}, 'd': {'g'}, 'f': {'d', 'g'}, 'e': {'d'}, 'a': {'c', 'b'}, 'c': {'f', 'e'}})

如果您更喜欢传统的方法，那么您可以添加一个if 条件

>>> answer = {}
>>> for line in f:
...     k,v = line.split(";")
...     if k in answer:
...         answer[k].add(v)
...     else:
...         answer[k] = {v}
... 
>>> answer
{'b': {'d'}, 'd': {'g'}, 'f': {'d', 'g'}, 'e': {'d'}, 'a': {'c', 'b'}, 'c': {'f', 'e'}}