Python 可变输入量

Question

我正在开发一个程序来确定一个图是否是强连接的。

我正在读取一系列行的标准输入。

这些行有两个或三个空格分隔的标记、源顶点和目标顶点的名称，以及一个可选的小数边权重。

输入可能如下所示：

'''
Houston Washington        1000
Vancouver Houston 300
Dallas Sacramento          800
Miami           Ames 2000
SanFrancisco LosAngeles
ORD PVD 1000
'''

如何读取此输入并将其添加到我的图表中？ 我相信我会使用这样的集合：

flights = collections.defaultdict(dict)

感谢您的帮助！

Answer 1

使用d 作为您的数据，您可以使用“\n”分割您的行，然后去除尾随空格并找到最后一次出现的。有了它，您可以对字符串进行切片以获取与其关联的名称和数字。

在这里，我将数据存储到字典中。您可以根据自己的需要进行修改！

使用正则表达式模块re.sub 删除多余的空格。

>>> import re
>>> d
'\nHouston Washington        1000\nVancouver Houston 300\nDallas Sacramento          800\nMiami           Ames 2000\nSanFrancisco LosAngeles\nORD PVD 1000\n'
>>>[{'Name':re.sub(r' +',' ',each[:each.strip().rfind(' ')]).strip(),'Flight Number':each[each.strip().rfind(' '):].strip()} for each in filter(None,d.split('\n'))]
[{'Flight Number': '1000', 'Name': 'Houston Washington'}, {'Flight Number': '300', 'Name': 'Vancouver Houston'}, {'Flight Number': '800', 'Name': 'Dallas Sacramento'}, {'Flight Number': '2000', 'Name': 'Miami Ames'}, {'Flight Number': 'LosAngeles', 'Name': 'SanFrancisco'}, {'Flight Number': '1000', 'Name': 'ORD PVD'}]

编辑：

为了匹配您的航班字典，

>>> flights={'Houston':{'Washington':''},'Vancouver':{'Houston':''}} #sample dict
>>> for each in filter(None,d.split('\n')):
...     flights[each.split()[0]][each.split()[1]]=each.split()[2]