阅读TSP文件Python

时间:2017-12-08 18:05:02

标签: python-2.7 traveling-salesman

我需要弄清楚如何读取文件名'berlin52.tsp'

的数据

这是我正在使用的格式

NAME: berlin52
TYPE: TSP
COMMENT: 52 locations in Berlin (Groetschel)
DIMENSION : 52
EDGE_WEIGHT_TYPE : EUC_2D
NODE_COORD_SECTION
1 565.0 575.0
2 25.0 185.0
3 345.0 750.0
4 945.0 685.0
5 845.0 655.0
6 880.0 660.0
7 25.0 230.0
8 525.0 1000.0
9 580.0 1175.0
10 650.0 1130.0

这是我目前的代码

# Open input file
infile = open('berlin52.tsp', 'r')

# Read instance header
Name = infile.readline().strip().split()[1] # NAME
FileType = infile.readline().strip().split()[1] # TYPE
Comment = infile.readline().strip().split()[1] # COMMENT
Dimension = infile.readline().strip().split()[1] # DIMENSION
EdgeWeightType = infile.readline().strip().split()[1] # EDGE_WEIGHT_TYPE
infile.readline()

# Read node list
nodelist = []
N = int(intDimension)
for i in range(0, int(intDimension)):
    x,y = infile.readline().strip().split()[1:]
    nodelist.append([int(x), int(y)])

# Close input file
infile.close()

代码应该在文件中读取,输出值为“1,2,3 ......”等的巡视列表,同时存储x和y值以计算距离。它至少可以收集标题。创建节点列表时会出现问题。

这是我得到的错误

ValueError: invalid literal for int() with base 10: '565.0'

我在这里做错了什么?

3 个答案:

答案 0 :(得分:1)

您正在将字符串“565.0”输入nodelist.append([int(x), int(y)])。 它告诉你它不喜欢它,因为该字符串不是整数。最后的.0使它成为一个浮动。

因此,如果您将其更改为nodelist.append([float(x), float(y)]),只作为一种可能的解决方案,那么您将看到问题消失。

或者,您可以尝试从字符串输入中删除或分离“.0”。

答案 1 :(得分:0)

这是TSPLIB格式的文件。要将其加载到python中,请查看可通过PyPiGithub

获得的python软件包tsplib95

文档在https://tsplib95.readthedocs.io/上可用

您可以将TSPLIB文件转换为networkx图,并从那里检索必要的信息。

答案 2 :(得分:0)

上面的代码有两个问题。我已经运行了代码,并在下面的行中发现了以下问题:

Dimension = infile.readline().strip().split()[1] 

这行应该是这样

`Dimension = infile.readline().strip().split()[2]`

而不是1,而是2,因为1 Dimension = :和2 Dimension = 52。 两者都是字符串类型。

第二个问题是第

N = int(intDimension)

将会是

 N = int(Dimension)

最后一行

for i in range(0, int(intDimension)):

只需使用

for i in range(0, N):

现在我想一切都会好起来的。