我对python来说比较新。 我想使用列名作为我的搜索数据从csv文件中读取特定列。有没有办法做到这一点?
答案 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."