将标题的文本文件读入字典

Question

我有目的地的文本文件，我需要将它们读入字典并根据出发地和目的地对其进行分类。

JFK MCO
ORD DEN
ORD HOU
DFW PHX
JFK ATL
ORD DFW
ORD PHX
ATL HOU
DEN PHX
PHX LAX
JFK ORD
DEN LAS
DFW HOU
ORD ATL
LAS LAX
ATL MCO
HOU MCO
LAS PHX
STL PDX

预期成绩：

{'JKF' : {'MCO', 'ATL','ORD'}, 'ORD' : {'DEN' , 'HOU' , 'DFW' , 'PHX' , 'ATL' ........}}

主要思想是起点在左侧，目的地在右侧。 将目的地分组到每个起点。

Answer 1

您可以遍历各行，拆分各行，然后使用dict.setdefault将目标存储到以原始位置为键的集合dict中：

d = {}
with open('file.txt') as file:
    for line in file:
        orig, dest = line.split()
        d.setdefault(orig, set()).add(dest)

d变为：

{'JFK': {'ORD', 'ATL', 'MCO'}, 'ORD': {'DEN', 'HOU', 'ATL', 'DFW', 'PHX'}, 'DFW': {'PHX', 'HOU'}, 'ATL': {'MCO', 'HOU'}, 'DEN': {'PHX', 'LAS'}, 'PHX': {'LAX'}, 'LAS': {'PHX', 'LAX'}, 'HOU': {'MCO'}, 'STL': {'PDX'}}

Answer 2

使用非常基本的for循环和dictdefaultdict可以很容易地完成此操作

from collections import defaultdict
d = defaultdict(set)
with open("filename.txt", "r") as f:
    for line in f: 
        start, place = line.split()
        d[start].add(place)

也可以使用csv模块来完成

from collections import defaultdict
import csv

d = defaultdict(set)
with open("filename.txt", "r") as f:
    r = csv.reader(f, delimiter=' ')
    for start, place in r: d[start].add(place)

当然，这导致defaultdict ，但你仍然可以使用它一样的字典，如果你想d是使用后的真实字典d = dict(d)

Answer 3

尝试这个

depart_place = ['JFK', 'ORD', 'ORD', 'DFW', 'JFK']
destination = ['MCO', 'DEN', 'HOU', 'PHX', 'ATL']

place_dict = {}

for place, dest in zip(depart_place, destination):
    if place in place_dict:
        place_dict[place].add(dest)
    else:
        place_dict[place] = {dest}

因此，从本质zip(depart_place, destination)我们正在遍历由zip(depart_place, destination)导致的出发地点和目的地(place, dest)元组。 然后，我们检查place已经在字典中，我们将追加destination它，否则，我们创建了一个新的key-value对的对place