如何刪除csv文件中的整個行並將更改保存在同一文件中？

Question

我是新的python，並嘗試修改csv文件，所以我將能夠根據給定的列表刪除特定字段的特定字段。 在我當前的代碼中，我得到了我想要刪除的行，但我無法刪除它並將更改保存在同一文件（替換）中。

import os, sys, glob 
import time ,csv 
# Open a file 
path = 'C:\\Users\\tzahi.k\\Desktop\\netzer\\' 
dirs = os.listdir(path) 
fileslst = [] 
alertsCode = ("42001", "42003", "42006","51001" , "51002" ,"61001" ,"61002","71001", 
      "71002","71003","71004","71005","71006","72001","72002","72003","72004", 
      "82001","82002","82003","82004","82005","82006","82007","83001","84001") 
# This would print the unnesscery codes 
for file in dirs: 
    if "ALERTS" in file.upper() : 
     fileslst.append(file) 
fileslst.sort() 

with open(fileslst[-1], 'rb') as csvfile: 
    csvReader = csv.reader(csvfile) 
    for row in csvReader: 
     for alert in alertsCode: 
      if any(alert in row[2] for s in alertsCode) : 
      print row 

有幫助嗎？

Answer 1

使用列表理解將所有行讀入列表中，並排除不需要的行。然後重寫行到重寫或替換該文件的內容，模式，文件w（寫模式）：

with open(fileslst[-1], 'rb') as csvfile: 
    csvReader = csv.reader(csvfile) 
    clean_rows = [row for row in csvReader if not any(alert in row[2] for alert in alertsCode)] 
    # csvfile.truncate() 

with open(fileslst[-1], 'wb') as csvfile: 
    csv_writer = csv.writer(csvfile) 
    csv_writer.writerows(clean_rows)