¿Cómo omitir los encabezados al procesar un archivo csv usando Python?

Question 1

Estoy usando el código mencionado a continuación para editar un csv usando Python. Las funciones llamadas en el código forman la parte superior del código.

Problema: quiero que el código mencionado a continuación comience a editar el csv desde la segunda fila, quiero que excluya la primera fila que contiene encabezados. En este momento, está aplicando las funciones solo en la primera fila y mi fila de encabezado está cambiando.

in_file = open("tmob_notcleaned.csv", "rb")
reader = csv.reader(in_file)
out_file = open("tmob_cleaned.csv", "wb")
writer = csv.writer(out_file)
row = 1
for row in reader:
    row[13] = handle_color(row[10])[1].replace(" - ","").strip()
    row[10] = handle_color(row[10])[0].replace("-","").replace("(","").replace(")","").strip()
    row[14] = handle_gb(row[10])[1].replace("-","").replace(" ","").replace("GB","").strip()
    row[10] = handle_gb(row[10])[0].strip()
    row[9] = handle_oem(row[10])[1].replace("Blackberry","RIM").replace("TMobile","T-Mobile").strip()
    row[15] = handle_addon(row[10])[1].strip()
    row[10] = handle_addon(row[10])[0].replace(" by","").replace("FREE","").strip()
    writer.writerow(row)
in_file.close()    
out_file.close()

Traté de resolver este problema inicializando row variable a 1 pero no funcionó.

Por favor, ayúdame a resolver este problema.

Question 2

Su reader variable es iterable, al recorrerla recupera las filas.

Para que salte un elemento antes de su ciclo, simplemente llame next(reader, None) e ignorar el valor de retorno.

También puede simplificar un poco su código; use los archivos abiertos como administradores de contexto para que se cierren automáticamente:

with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
   reader = csv.reader(infile)
   next(reader, None)  # skip the headers
   writer = csv.writer(outfile)
   for row in reader:
       # process each row
       writer.writerow(row)

# no need to close, the files are closed automatically when you get to this point.

Si desea escribir el encabezado en el archivo de salida sin procesar, eso también es fácil, pase la salida de next() a writer.writerow():

headers = next(reader, None)  # returns the headers or `None` if the input is empty
if headers:
    writer.writerow(headers)

Question 3

Otra forma de resolver esto es usar la clase DictReader, que “salta” la fila del encabezado y la usa para permitir la indexación con nombre.

Dado “foo.csv” de la siguiente manera:

FirstColumn,SecondColumn
asdf,1234
qwer,5678

Use DictReader así:

import csv
with open('foo.csv') as f:
    reader = csv.DictReader(f, delimiter=",")
    for row in reader:
        print(row['FirstColumn'])  # Access by column header instead of column number
        print(row['SecondColumn'])

Question 4

Haciendo row=1 no cambiará nada, porque simplemente lo sobrescribirá con los resultados del ciclo.

Quieres hacer next(reader) para saltar una fila.

Question 5

Simplemente iterar una vez con next()

with open(filename) as file:

    csvreaded = csv.reader(file)
    header = next(csvreaded)

    for row in csvreaded:
        empty_list.append(row) #your csv list without header

o usar [1:] al final del objeto del lector

with open(filename) as file:

    csvreaded = csv.reader(file)
    header = next(csvreaded)

    for row in csvreaded[1:]:
        empty_list.append(row) #your csv list without header

Question 6

Inspirado en la respuesta de Martijn Pieters.

En caso de que solo necesite eliminar el encabezado de la csv archivo, puede trabajar de manera más eficiente si escribe usando la biblioteca estándar de E/S de archivos de Python, evitando escribir con la biblioteca de Python CSV:

with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
   next(infile)  # skip the headers
   outfile.write(infile.read())

¿Ha sido útil esta solución?