Kopieren bestimmter Zeilen von CSV in CSV in Python 2.7

Question

Bis jetzt habe ich versucht, bestimmte Zeilen einschließlich Header aus der ursprünglichen CSV-Datei in eine neue zu kopieren. Sobald ich meinen Code ausführte, konnte ich jedoch ein riesiges Dokument kopieren.

Dies ist eine der Optionen, die ich bisher versucht haben, die in der Nähe von der Lösung zu sein scheint:

import csv 
with open('D:/test.csv', 'r') as f,open('D:/out.csv', 'w') as f_out: 
    reader = csv.DictReader(f) 
    writer = csv.writer(f_out) 
    for row in reader: 
     if row["ICLEVEL"] == "1": 
      writer.writerow(row) 

Die Sache ist, dass ich nur die Zeilen kopieren, wo Wert von „ICLEVEL“ (Kopfzeilenname) ist gleich "1".

Hinweis: test.csv ist sehr große Datei, und ich kann nicht alle Header-Namen im Writer fest codieren.

Jede Demonstration der pythonischen Art, dies zu tun, wird sehr geschätzt. Vielen Dank.

Answer 1

Ihr row ist ein Wörterbuch. CSV Writer kann keine Wörterbücher schreiben. Wählen Sie die Werte aus dem Wörterbuch und schreibe sie nur:

writer.writerow(reader.fieldnames) 
for row in reader: 
    if row["ICLEVEL"] == "1": 
    values = [row[field] for field in reader.fieldnames] 
    writer.writerow(values) 

Answer 2

writer.writerow erwartet eine Folge (ein Tupel oder Liste). Sie können DictWriter verwenden, die eine dict erwartet.

import csv 
with open('D:/test.csv', 'r') as f, open('D:/out.csv', 'w') as f_out: 
    reader = csv.DictReader(f) 
    writer = csv.DictWriter(f_out, fieldnames=reader.fieldnames) 
    writer.writeheader() # For writing header 
    for row in reader: 
     if row['ICLEVEL'] == '1': 
      writer.writerow(row) 

Answer 3

Ich würde tatsächlich nutzen Pandas, nicht einen CSV-Leser:

import pandas as pd 

df=pd.read_csv("D:/test.csv") 
newdf = df[df["ICLEVEL"]==1] 
newdf.to_csv("D:/out.csv",index=False) 

Der Code ist viel kompakter.