Saltar al contenido

Leer columnas específicas con pandas u otro módulo de Python

Solución:

Una forma sencilla de hacer esto es usar el pandas biblioteca como esta.

import pandas as pd
fields = ['star_name', 'ra']

df = pd.read_csv('data.csv', skipinitialspace=True, usecols=fields)
# See the keys
print df.keys()
# See content in 'star_name'
print df.star_name

El problema aquí era el skipinitialspace que eliminan los espacios en el encabezado. Entonces ‘star_name’ se convierte en ‘star_name’

De acuerdo con la última documentación de pandas, puede leer un archivo csv seleccionando solo las columnas que desea leer.

import pandas as pd

df = pd.read_csv('some_data.csv', usecols = ['col1','col2'], low_memory = True)

Aquí usamos usecols que lee solo las columnas seleccionadas en un marco de datos.

Estamos usando low_memory para que procesemos internamente el archivo en trozos.

¡Haz clic para puntuar esta entrada!
(Votos: 0 Promedio: 0)



Utiliza Nuestro Buscador

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *