如何从csv中读取特定列?

时间:2017-07-06 04:12:34

标签: python-2.7

我对python来说比较新。 我想使用列名作为我的搜索数据从csv文件中读取特定列。有没有办法做到这一点?

1 个答案:

答案 0 :(得分:0)

import csv
import sys

print "What column are you searching for? "
userinput = sys.stdin.readline().rstrip();

with open("my.csv", "rb") as f:
    reader = csv.reader(f)
    header = reader.next()
    try:
        pos = header.index(userinput)
        print "Values in column " + userinput + ": "
        for row in reader:
            print row[pos]
    except:
        print "column not found."