如何使用Python腳本從CSV文件過濾

Question

我有abx.csv文件有三列。 我想過濾具有Application as 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 （修改現有文件）

Answer 1

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)

Answer 2

您應該使用不同的輸出文件名。 即使您希望名稱相同，也應使用一些臨時名稱，最后重命名文件。 否則你必須先將文件讀入內存

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