2017-07-06 231 views

回答

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."