[問題] 如何搜尋CSV的資料內容

[superge (Ge肥貓)]

各位好

我想要搜尋CSV檔裡的內容，因為資料量大，所以想寫一個程式來搜尋

假設我知道裡面幾個欄位的資料，我想要藉由這幾個欄位的資料來搜尋與這幾個欄位

相關的資料

好比說我知道a欄位的資料，我只需要輸入a欄位就可以找到跟他相關的資訊

以下是我寫的程式碼:

# -*- coding: utf-8 -*-
import csv
f = open('ac2.csv','r')
a = input('serch')
for d in csv.DictReader(f, ["FrameNo", "FilmPicNo", "OriginalFileName"]):

        if a == d['FrameNo']:
                print d['OriginalFileName']
        else:
                print 'haha loser UCCU'
        print d['FilmPicNo']
f.close()

我這樣寫沒辦法讓 a == d

所以一直印出 else 的結果和 Print d['FilmPicNo']

我需要怎麼改善我的程式碼??

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 220.139.22.77
※ 文章網址: http://www.ptt.cc/bbs/Python/M.1404281456.A.C5E.html
推 Seudo: a轉字串
→ monkeytiger:python 3以前的版本raw_input() vs. input()