Wie ersetzt man eine Spalte in einer CSV-Datei in Python?

Question

Ich habe 2 csv-Dateien. Ich muss eine Spalte in einer Datei durch eine Spalte aus der anderen Datei ersetzen, aber sie müssen nach einer ID-Spalte sortiert bleiben.

Hier ist ein Beispiel:

file1:

ID, transect, 90mdist          
1, a, 10,             
2, b, 20,             
3, c, 30,  

file2:

ID, transect, 90mdist         
1, a, 50             
2, b, 70              
3, c, 90   

im Grunde habe ich eine neue Datei mit dem richtigen 90mdist und ich brauche es in die alte Datei einfügen aber es muss mit der gleichen ID # übereinstimmen.

Es ist mein Verständnis, dass Python CSV-Dateien als Zeichenfolge behandelt. also kann ich entweder ein Wörterbuch verwenden oder die Daten in eine Liste konvertieren und dann ändern? Welcher Weg ist am besten?

Jede Hilfe würde sehr geschätzt werden !!

Answer 1

Die CSV Module in der Python-Bibliothek ist, was Sie hier brauchen.

Es ermöglicht das Lesen und Schreiben von CSV-Dateien, das Behandeln von Zeilen mit Tupeln oder Listen von Elementen.

Lesen Sie einfach die Datei mit den korrigierten Werten ein, speichern Sie sie in einem Wörterbuch mit der ID der Linie.

Dann lesen Sie in der zweiten Datei, ersetzen die relevante Spalte mit den Daten aus dem Diktat und schreiben Sie in eine dritte Datei.

Fertig.

Answer 2

Sobald Sie Ihre CSV-Listen haben, wäre eine einfache Möglichkeit, eine Spalte in einer Matrix durch eine andere zu ersetzen, die Matrizen zu transponieren, die Zeile zu ersetzen und dann Ihre bearbeitete Matrix zurück zu transponieren. Hier ist ein Beispiel mit Ihren Daten:

csv1 = [['1', 'a', '10'], ['2', 'b', '20'], ['3', 'c', '30']] 
csv2 = [['1', 'a', '50'], ['2', 'b', '70'], ['3', 'c', '90']] 

# transpose in Python is zip(*myData) 
transposedCSV1, transposedCSV2 = zip(*csv1), zip(*csv2) 
print transposedCSV1 
>>> [['1', '2', '3'], ['a', 'b', 'c'], ['10', '20', '30']] 

csv1 = transposedCSV1[:2] + [transposedCSV2[2]] 
print csv1 
>>> [['1', '2', '3'], ['a', 'b', 'c'], ['50', '70', '90']] 

csv1 = zip(*csv1) 
print csv1 
>>> [['1', 'a', '50'], ['2', 'b', '70'], ['3', 'c', '90']] 

Answer 3

Wenn Sie nur diese als einmaligen tun, warum mit Python überhaupt die Mühe machen? Excel oder OpenOffice Calc öffnet die zwei CSV-Dateien für Sie, dann können Sie einfach die Spalte ausschneiden und einfügen.

Wenn die zwei Listen der IDs nicht genau die gleichen sind, würde ein einfaches VB-Makro es für Sie tun.

Answer 4

Versuchen Sie folgendes:

from __future__ import with_statement 

import csv 

def twiddle_csv(file1, file2): 
    def mess_with_record(record): 
     record['90mdist'] = 2 * int(record['90mdist']) + 30 
    with open(file1, "r") as fin: 
     with open(file2, "w") as fout: 
      fields = ['ID', 'transect', '90mdist'] 
      reader = csv.DictReader(fin, fieldnames=fields) 
      writer = csv.DictWriter(fout, fieldnames=fields) 
      fout.write(",".join(fields) + '\n') 
      reader.next() # Skip the column header 
      for record in reader: 
       mess_with_record(record) 
       writer.writerow(record) 

if __name__ == '__main__': 
    twiddle_csv('file1', 'file2') 

Ein paar Einschränkungen:

• DictReader die erste Zeile als Daten zu verwenden scheint, auch wenn es die Spiele Felder aus. Rufen Sie reader.next() auf, um zu überspringen.
• Datenzeilen dürfen keine abschließenden Kommas enthalten. Sie werden als leere Spalten interpretiert.
• DictWriter scheint die Spaltenüberschriften nicht auszugeben. DIY.