es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Intentando agregar una columna de índice a un archivo CSV.

Estaba tratando de agregar una columna de índice a un archivo CSV. No puedo usar pandas ya que mi archivo es demasiado grande y con pandas no puede leer todos los datos.

Intenté hacerlo de esta manera:

with open('test.csv') as infile, open('sample_.csv', 'w') as outfile:
    for idx, line in enumerate(infile):
        outfile.write(f'{idx}   {line}')

Esto me da una columna, sin embargo, la columna de índice se fusiona con la columna existente.

Ver imagen aquí

Tags:  ,

Answer

  1. Avatar for davy.ai

    Para evitar la fusión de la columna de índice con la primera columna ya existente, puede modificar el código de la siguiente manera:

    with open('test.csv') as infile, open('sample_.csv', 'w') as outfile:
        outfile.write('index,' + infile.readline()) # escribir encabezados con la nueva columna de índice
        for idx, line in enumerate(infile):
            outfile.write(f'{idx},{line}') # escribir línea con la nueva columna de índice
    

    Al agregar una nueva columna de encabezado ‘index’ en el archivo de salida, la columna de índice no se fusionará con la primera columna del archivo CSV original. El código actualizado le dará un archivo CSV de salida con dos columnas: ‘index’ y las columnas existentes del archivo original.

Comments are closed.