Hoy vamos a ver un ejemplo simple con Python y los archivos CSV. Veremos cómo convertir un CSV a un arreglo.

Primero vamos a convertir cada columna del archivo a un diccionario de Python, y después vamos a agregar ese diccionario a nuestra lista.

En el ejemplo te mostraré cómo leer datos enteros y flotantes, así como cadenas. De este modo tú podrás leer cualquier archivo separado por comas para tenerlo como array en Python.

Nota: yo decidí usar un diccionario, tú podrías usar una lista de listas o cualquier otra cosa.

Explicación del algoritmo

Vamos a abrir y leer el archivo línea por línea (omitiendo el encabezado si es necesario). En cada paso, tendremos una línea del CSV.

Ahora convertiremos esa línea a arreglo con split, y ya  tendremos las columnas separadas. Luego transformamos cada valor a su tipo en caso de ser necesario, y lo asignamos al diccionario.

Cuando tenemos el diccionario, lo agregamos al arreglo con append. Y finalmente ya tendremos el CSV cargado a nuestro arreglo con Python.

El archivo CSV

Para este ejemplo tengo un archivo de alumnos en donde la primera línea es el encabezado y luego tenemos, en orden, el nombre, la edad, la altura y la matrícula.

See the gist on github.

(dale a view raw al gist para que veas el archivo real).

El orden de las columnas importa, ya que los vamos a leer así más adelante.

CSV a lista de Python

Entonces abrimos el archivo. Definimos el separador y la lista donde vamos a tener los datos:

See the gist on github.

Luego iteramos el archivo línea por línea, quitamos el salto de línea con rstrip, convertimos y asignamos. La definición del diccionario está junto con la llamada a append en la línea 8.

See the gist on github.

Fíjate en que estoy usando int y float para convertir las cadenas a números. Finalmente encerramos todo el comportamiento en una función:

See the gist on github.

Y ya podemos invocar a esta función desde cualquier lugar, pasándole el nombre del archivo.

Poniendo todo junto

Entonces con el siguiente CSV:

See the gist on github.

Podemos usar el siguiente código para leer el archivo e imprimir cada dato por separado:

See the gist on github.

La salida es correcta:

En este caso solo estoy imprimiendo los datos pero tú puedes hacer cualquier otro tipo de cosas con ellos.

Y con este ejemplo espero que quede claro cómo leer un CSV en Python para convertirlo a un arreglo.