2014-02-15 65 views
0

考慮採用以下格式的.csv文件:蟒蛇:按行名從CSV文件中提取值

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

我需要檢索第二列中的值具有「約翰」的所有行寫在第一列。我想用python腳本來做。我對Python非常陌生,所以我在問怎麼做呢?

+0

看看在[CSV模塊(HTTP ://docs.python.org/2/library/csv.html)。 – ford

+0

輸入CSV文件中的逗號後面是否有空格? –

+0

@MartijnPieters,你是對的,沒有空格。我編輯了這個問題。 – Subway

回答

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.0版本是這樣的:

import csv 

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

你的假設是正確的。謝謝。 – Subway