Python und CSV; Wie schneidet man alle Werte in einer Spalte ab?

Question

eine einfache CSV-Datei wie folgt gegeben:

Django,Gunslinger,101-707 
KingSchultz,Dentist,205-707 
Tatum,Marshall,615-707 
Broomhilda,Wife,910-707 
...,...,... 

Wie Sie alle Werte in der letzten Spalte gestutzt, so dass nur die ersten drei Ziffern bleiben? (Nicht verwandt: so können sie in mathematischen Operationen verwendet werden)

Wunsch CSV:

Django,Gunslinger,101 
KingSchultz,Dentist,205 
Tatum,Marshall,615 
Broomhilda,Wife,910 
...,...,... 

Hier ist, was ich bisher versucht:

import csv 
import re 
r = csv.reader(open(input.csv)) 
for row in r: 
    re.sub('\-.*', '', row[3]) 
writer = csv.writer(open('output.csv', 'w')) 
writer.writerow(row) 

ich die regex in re.sub prüft haben funktioniert richtig. Habe Dutzende von Varianten probiert, viele Stunden lang gesucht, aber nicht die gewünschte Ausgabe bekommen.

Answer 1

Ohne Verwendung re Modul

import csv 

r = csv.reader(open("sample.csv", "rb")) 
writer = csv.writer(open("output.csv", "wb")) 

for row in r: 
    row[2] = row[2][:3] 
    writer.writerow(row) 

Wie @TigerRedMike wies darauf hin, in Python 3.X, statt 'rb' und: richtig zu arbeiten (python3)

Hier Ihren Code geändert 'wb', 'r' und 'w' sollten jeweils zum Lesen und Schreiben der Dateien verwendet werden.

Answer 2

re.sub gibt die Zeichenfolge mit der Ersetzung zurück. es hat keinen Einfluss auf das dritte Argument selbst

Answer 3

Das erste, was re.sub gibt einen neuen Wert zurück, es ersetzt nicht den Wert in der Zeile.

zweitens row ist eine laufende Variable, sollten Sie dies innerhalb der Schleife behandeln.

import csv 
import re 
r = csv.reader(open('.../test.csv')) 
with open('.../test2.csv', 'w') as csvfile: 
    writer = csv.writer(csvfile, delimiter=',') 
    for row in r: 
     row[2] = re.sub('\-.*', '', row[2]) 
     writer.writerow(row) 

Answer 4

import csv 
import re 

r = csv.reader(open('input.csv')) 
writer = csv.writer(open('/tmp/output.csv', 'w')) 

for row in r: 
    tmp = re.sub('\-.*', '', row) 
    writer.writerow(tmp)