用Python解析CSV文件

Question

我有一個CSV文件，包含大約30個標題（列）和2000行。

HeaderOne | HeaderTwo | HeaderThree
dataRowOne |  dataRowOne | dataRowOne
dataRowTwo |  dataRowTwo | dataRowTwo

我想使用Python搜索字符串，然后輸出該行。 舉例來說，例如我搜索了'cocaColaIsTheBest'，它在單元格E2018中，我希望Python打印出2018年行，其標題位於其上方。

到目前為止，我已經：

import csv

myExSpreadSheet = csv.reader(open('Simon.csv', 'rb'))

for row in myExSpreadSheet:
    if 'cocaColaIsTheBest' in row:
        print (row)

這將在字典中打印該行； 我也希望它打印標題。

如何打印所有標題？

以及如何打印特定的標題？

Answer 1

標頭是第一行； 首先捕獲那些：

with open('Simon.csv', 'rb') as csvfile:
    myExSpreadSheet = csv.reader(csvfile)
    headers = next(myExSpreadSheet, None)  # grab first row

    for row in myExSpreadSheet:
        if 'cocaColaIsTheBest' in row:
            print headers
            print row

Answer 2

您確定使用DictReader更好嗎？ 然后，標題與它們相應的單元格相關聯，並且可以根據需要設置格式。