Pandas solo lee columnas específicas de CSV

Pulamolu Sai Mohan 21 junio 2023
  1. Use la función read_csv() y pase una lista de nombres de columna para leer
  2. Use la función read_csv() y pase la lista de números de columna para leer
Pandas solo lee columnas específicas de CSV

Este tutorial discutirá diferentes métodos para leer columnas específicas de un archivo csv usando Pandas. Usaremos el siguiente archivo CSV para la demostración.

Curso Mentor precio $
0 Pitón roberto 199
1 Chispa - chispear Isabel 299
2 Java Nolan 99
3 JavaScript cris 250
4 C# johnson 399

Use la función read_csv() y pase una lista de nombres de columna para leer

La biblioteca de Pandas proporciona la función read_csv(). El read_csv() se utiliza para leer un archivo de valores separados por comas (csv) en DataFrame.

El read_csv() toma la ruta del archivo csv y tiene más de 40 parámetros opcionales. Para este tutorial, concentrémonos en el parámetro usecols.

Sintaxis - pd.read_csv():

pd.read_csv(filepath, usecols)

Parámetros

  • filepath - La ruta o ubicación del archivo csv que se va a cargar.
  • usecols - Lista de columnas específicas a leer.

Devoluciones

  • DataFrame

Siga los pasos a continuación para usar la función read_csv() y pase la lista de nombres de columna para leer.

  • Importar biblioteca de pandas.
  • Pase la ruta del archivo csv y la lista de columnas para leer al método read_csv().
  • El método read_csv() devolverá un marco de datos con esas columnas específicas.
  • Imprime el DataFrame.

El siguiente código es la implementación del enfoque anterior.

# import the pandas
import pandas as pd

# Reading the specific columns from csv
df = pd.read_csv("data.csv", usecols=["Course", "price$"])

# printing the dataframe
print(df)

Producción :

       Course  price$
0      Python     199
1       Spark     299
2        Java      99
3  JavaScript     250
4          C#     399

Use la función read_csv() y pase la lista de números de columna para leer

También podemos usar los números de columna para read_csv() para columnas específicas de solo lectura.

  • Importar biblioteca de pandas.
  • Pase la ruta del archivo csv y la lista de números de columnas para leer al método read_csv().
  • El método read_csv() devolverá un marco de datos con esas columnas específicas.
  • Imprime el DataFrame.

El siguiente código es la implementación del enfoque anterior.

# import the pandas
import pandas as pd

# Reading the specific columns from csv
df = pd.read_csv("data.csv", usecols=[2, 3])

# printing the dataframe
print(df)

Producción :

     Mentor  price$
0     Robert     199
1  Elizibeth     299
2      Nolan      99
3      Chris     250
4    johnson     399

Artículo relacionado - Pandas DataFrame