如何從csv讀取特定列？

Question

我對python比較陌生。 我想使用列名作爲我的搜索數據從csv文件中讀取特定列。有沒有辦法做到這一點 ？

Answer 1

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