python：按行名从csv文件中提取值

Question

考虑使用以下格式的.csv文件：

John,29,21,,DF,
Sara,23,51,,DF,
John,34,27,,ER,
John,76,29,,TY,
Sara,87,93,,SAD,

我需要在第二列中检索第一列中写有“John”的所有行的值。我想用python脚本来做。我对python很新，所以我问这怎么办？

Answer 1

csv module使这个变得微不足道：

import csv

with open(inputfilename, 'rb') as infh:
    reader = csv.reader(infh)
    for row in reader:
        if row[0] == 'John':
            print row[1]

这假定您使用的是Python 2. Python 3版本如下：

import csv

with open(inputfilename, newline='') as infh:
    reader = csv.reader(infh)
    for row in reader:
        if row[0] == 'John':
            print(row[1])