在python中使用CSV文件打印列

Question

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

这是我的 csv 文件的布局，其中 1,2,etc 是 CID（客户 ID） 1,Name,Street addr2,Postcode,X,Y 2,Name,Street_addr2,Pcode,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 并且程序读取坐标的 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']