Python – Leer columnas CSV en la lista

En este artículo, leeremos los datos de un archivo CSV en una lista. Usaremos la biblioteca de panda para leer los datos en una lista. 

Archivo utilizado: archivo .

Método 1: usar pandas

Aquí tenemos la función read_csv() que ayuda a leer el archivo CSV simplemente creando su objeto. El nombre de la columna se puede escribir dentro de este objeto para acceder a una columna en particular, lo mismo que hacemos para acceder a los elementos de la array. La biblioteca de Pandas tiene una función llamada tolist() que convierte los datos en una lista que se puede usar según nuestros requisitos. Entonces, usaremos esto para convertir los datos de la columna en una lista. Finalmente, imprimiremos la lista.

Acercarse:

• Importar el módulo.
• Leer datos del archivo CSV.
• Conviértalo en la lista.
• Imprime la lista.

A continuación se muestra la implementación:

# importing module
from pandas import *
 
# reading CSV file
data = read_csv("company_sales_data.csv")
 
# converting column data to list
month = data['month_number'].tolist()
fc = data['facecream'].tolist()
fw = data['facewash'].tolist()
tp = data['toothpaste'].tolist()
sh = data['shampoo'].tolist()
 
# printing list data
print('Facecream:', fc)
print('Facewash:', fw)
print('Toothpaste:', tp)
print('Shampoo:', sh)

Producción:

Método 2: Usando el módulo csv

En este método importaremos la biblioteca csv y abriremos el archivo en modo lectura, luego usaremos la función DictReader() para leer los datos del archivo CSV. Esta función es como un lector normal, pero asigna la información a un diccionario cuyas claves vienen dadas por los nombres de las columnas y todos los valores como claves. Crearemos listas vacías para que podamos almacenar los valores en ella. Finalmente, accedemos a los valores clave y los agregamos a las listas vacías e imprimimos esa lista.

# importing the module
import csv
 
# open the file in read mode
filename = open('company_sales_data.csv', 'r')
 
# creating dictreader object
file = csv.DictReader(filename)
 
# creating empty lists
month = []
totalprofit = []
totalunit = []
 
# iterating over each row and append
# values to empty list
for col in file:
    month.append(col['month_number'])
    totalprofit.append(col['moisturizer'])
    totalunit.append(col['total_units'])
 
# printing lists
print('Month:', month)
print('Moisturizer:', totalprofit)
print('Total Units:', totalunit)

Producción: