如何使用Python脚本过滤CSV文件

j2qf4p5b  于 2022-12-15  发布在  Python
关注(0)|答案(2)|浏览(178)

我有一个包含三列的abx.csv文件。我想筛选包含Application的数据作为Central,并将其写入同一个.csv文件

User ID   Name        Application  

001       Ajohns      ABI
002       Fjerry      Central
900       Xknight     RFC
300       JollK       QDI
078       Demik       Central

我需要在同一个.csv文件中的三列中写入User ID,Name,Apllication(修改现有文件)

rlcwz9us

rlcwz9us1#

import csv
reader = csv.reader(open(r"abx.csv"),delimiter=' ')
filtered = filter(lambda p: 'Central' == p[2], reader)
csv.writer(open(r"abx.csv",'w'),delimiter=' ').writerows(filtered)
deyfvvtc

deyfvvtc2#

您应该使用不同的输出文件名。即使您希望名称相同,您应该使用一些临时名称,最后重命名文件。否则,您必须先将文件读取到内存中

import csv
with open('infile','r') as fin, open ('outfile','w') as fout:
    writer = csv.writer(fout, delimiter=' ')            
    for row in csv.reader(fin, delimiter=' '):
        if row[2] == 'Central':
             writer.writerow(row)

相关问题