检查csv文件中的特定列是否包含数据

Question

我需要检查我行的第三列包含 2016 年。

到目前为止我的代码：

    import csv
    import pandas
    from docutils.utils import column_indices
    from bsddb.dbtables import contains_metastrings
    file = open("Advertising.csv")

    #write new file
    c = csv.writer(open("boop.csv", "wb"))
    #read in original file
    readCSV = csv.reader(file,delimiter= ",")


    for row in readCSV :
        print "something"
        #if contains(2016)
        if readCSV.index_col(2).Contains('2016') :
            print "2016 spotted"
            c.writerow(row)

    file.close()

检查第三列的行是错误的，并产生错误：AttributeError: '_csv.reader' object has no attribute 'index_col' 谁能帮忙解决这个问题？

Answer 1

你不需要pandas这个。csv模块会很好

import csv

with open("file.csv", "rb") as f:
    csvreader = csv.reader(f, delimiter=",")
    for row in csvreader:
        if "2016" in row[2]:
            print "2016 spotted"