如何将文件中的一行解析为多个列表

Question

我有一个像这样的文件。你知道吗

1,2,Room1 # first is the current room, the rest of the numbers the neighbours - maximum 4, and the last the name of the room
2,1,5,Room2
5,2,332,Room3

。。。。等等

我想解析它以便使用BFS算法。我试过：

with open( "data.txt", "r" ) as input_data:
  input_list= [map(int,num.split()) for num in input_data.readlines()]

我想要一本这样的字典，以便应用BFS：

data = { '1':['2'], '2':['1','5'] .....}

你知道怎么解析吗？你知道吗

Answer 1

你在找这样的东西吗？你知道吗

with open( "data.txt", "r" ) as input_data:
    input_list= input_data.readlines()

new_dict = {}
for mem in input_list:
    a = mem.split(',')
    new_dict[a[0]] = a[1:len(a) - 1]

print new_dict
//{'1': ['2'], '2': ['1', '5'], '5': ['2', '332']}

可能会更优化

Answer 2

您的输出包含字符串，因此可以跳过int调用。你知道吗

如果第一个元素是键，其余元素是值（忽略房间名称），那么一个dict理解就足够了：

with open( "data.txt", "r" ) as input_data:
    data = {elems[0]: elems[1:-1] 
            for line in input_data for elems in (line.split(','),)}

for elems in (line.split(','),)部分只为line.split(',')的输出分配一个值；它绕过了理解语法的限制。你知道吗

因为这是逗号分隔的数据，所以我在这里使用^{} module：

import csv

with open( "data.txt", "rb" ) as input_data:
    reader = csv.reader(input_data)
    data = {row[0]: row[1:-1] for row in reader}

如您所见，这大大简化了对每一行的处理。你知道吗

如果需要，添加int()调用：

data = {int(row[0]): map(int, row[1:-1]) for row in reader}