Lea un CSV en la lista de listas en Python

En este artículo, veremos cómo leer archivos CSV en una lista de listas en Python.

Método 1: Usando el módulo CSV

• Podemos leer los archivos CSV en diferentes estructuras de datos como una lista, una lista de tuplas o una lista de diccionarios.
• Podemos usar otros módulos como pandas, que se usan principalmente en aplicaciones de ML y cubren escenarios para importar contenido CSV a una lista con o sin encabezados.

Ejemplo 1:

En este ejemplo, estamos leyendo un archivo CSV y convirtiendo la string en la lista.

 

import csv
  
with open('sample.csv', 'r') as read_obj:
  
    # Return a reader object which will
    # iterate over lines in the given csvfile
    csv_reader = csv.reader(read_obj)
  
    # convert string to list
    list_of_csv = list(csv_reader)
  
    print(list_of_csv)

Producción:

[[‘ENE’, 34, 360, 417], [‘FEB’, 31, 342, 391], [‘MAR’, 36, 406, 419], [‘ABR’, 34, 396, 461],

 [‘MAY’, 36, 420, 472], [‘JUN’, 43, 472, 535], [‘JUL’, 49, 548, 622], [‘AGO’, 50, 559, 606], 

 [‘SEP’, 40, 463, 508], [‘OCT’, 35, 407, 461], [‘NOV’, 31, 362, 390], [‘DEC’, 33, 405, 432]]

Ejemplo 2:

En este ejemplo, estamos leyendo un archivo CSV e iterando sobre líneas en el CSV dado.

 

import csv
  
with open('example.csv') as csvfile:
    
    # Return a reader object which will
    # iterate over lines in the given csvfile.
    readCSV = csv.reader(csvfile, delimiter=',')
    for row in readCSV:
        print(row)
        print(row[0])
        print(row[0], row[1], row[2],)
        print("\n")

Producción:

 

Método 2: usar pandas

Puede usar la biblioteca pandas para esto, que tiene un método incorporado para convertir valores en una lista. La propiedad Pandas.values ​​se usa para obtener un numpy.array y luego usa la función tolist() para convertir esa array en una lista.

Nota: Para obtener más información, consulte Leer CSV en la lista usando Pandas

# app.py
  
import pandas as pd
  
# Creating Dictionary
dict = {
    'series': ['Friends', 'Money Heist', 'Marvel'],
    'episodes': [200, 50, 45],
    'actors': [' David Crane', 'Alvaro', 'Stan Lee']
}
  
# Creating Dataframe
df = pd.DataFrame(dict)
print(df)

Producción: