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

7xzttuei  于 2023-04-18  发布在  其他
关注(0)|答案(1)|浏览(118)

我需要检查CSV文件中我的行的第三列是否包含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'

有人能帮忙吗?

nbnkbykc

nbnkbykc1#

对此,您不需要pandascsv模块就可以了

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"

相关问题