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：＆＃39; _csv.reader＆＃39;对象没有属性＆＃39; index_col＆＃39; 任何人都可以帮忙吗？

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"

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

1 个答案: