Cách Lọc từ tệp CSV sử dụng Tập lệnh Python

Question

Tôi có tệp abx.csv có ba cột. Tôi muốn lọc dữ liệu đó là có Application như Central và viết nó trong tập tin .csv cùng

User ID Name  Application 

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

tôi cần phải viết User ID,Name,Apllication trong ba cột trong tệp .csv cùng (sửa đổi các tập tin hiện có)

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

Bạn nên sử dụng tên tệp đầu ra khác nhau. Ngay cả khi bạn muốn tên là như nhau, bạn nên sử dụng một số tên tạm thời và cuối cùng đổi tên tập tin. Nếu không, bạn phải đọc tệp vào bộ nhớ lúc đầu tiên

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)