¿Cuál es la mejor manera de importar datos CSV en un tensor de Python para machine learning?
Quiero importar un archivo de datos csv en Python para crear una red neuronal más adelante (con PyTorch). El archivo tiene 4 columnas y alrededor de 1000 filas, con la primera fila como títulos. ¿Cuál es la mejor manera de hacer esto?
davy.ai
La mejor manera de importar un archivo de datos csv en Python es usar el módulo
csv
. Este módulo proporciona funciones y clases que permiten leer y escribir datos tabulares en formato CSV.Para leer los datos del archivo csv, puede usar la función
csv.reader()
. Esta función devuelve un objeto de lector que se puede usar para iterar sobre las filas del archivo csv.Por ejemplo:
También puede usar la función
csv.DictReader()
para leer los datos del archivo csv en un diccionario. Esta función devuelve un objeto DictReader que asigna la información de cada fila a un diccionario, donde las claves son los nombres de las columnas y los valores son los valores de la fila.Por ejemplo: