Question

我使用python来解决TSP问题。我已经设法对TSP进行排序，但是我无法让python读取输入中的特定列。

这是我的csv文件的布局，其中1,2等是CID（客户ID） 1，名称，街道地址2，邮政编码，X，Y 2，名称，Street_addr2，P代码，X，Y

我需要我的程序读取输入上的CID但打印并附加x，y以用于数学计算。

我是python的新手所以任何提示都会受到赞赏。

由于

 print "Route Calculator"
cur = (0, 0)
xy = []
route = [0]
n = input("Enter number of destinations to visit:")
a = range(1, n + 1)
for i in range(n):
    x, y = input("Enter co-ordinates as x,y:")
    xy.append((x, y))

很抱歉早期帖子中缺少代码而且含糊不清。我基本上希望x，y输入来自CSV，如果这甚至可能的话。而不是x，y输入 - 客户需要输入CID并且程序读取cos的csv文件

再次感谢

Answer 1

示例代码：

import csv

def Cid2xy(csv_file, cid):
    reader = csv.reader(csv_file)
    while True:
        try:
            line = reader.next()
            if line[0] == cid:
                return line[4:6]
        except StopIteration:
            return None

CSV文件内容：

1,Name1,Street addr1,Postcode,100,100
2,Name2,Street_addr2,Pcode,200,200

测试输出（注意所有值都是字符串）：

>> cid.Cid2xy(csv_file, '1')
['100', '100']

在python中使用CSV文件来打印列

1 个答案: