Pandas: remova uma coluna específica ao importar um arquivo csv

Você pode usar a seguinte sintaxe básica para remover uma coluna específica ao importar um arquivo CSV para um DataFrame do pandas:

 df = pd. read_csv (' basketball_data.csv ', usecols= lambda x:x != ' rebounds ')

Este exemplo específico lerá cada coluna de um arquivo CSV chamado basquete_data.csv em um DataFrame do pandas, exceto a coluna chamada rebotes .

O exemplo a seguir mostra como usar essa sintaxe na prática.

Exemplo: Remover uma coluna específica ao importar um arquivo CSV para o Pandas

Digamos que temos o seguinte arquivo CSV chamado basket_data.csv :

Podemos usar a seguinte sintaxe para importar o arquivo CSV para o pandas e remover a coluna chamada bounces durante a importação:

 import pandas as pd

#import all columns except 'rebounds' into DataFrame
df = pd. read_csv (' basketball_data.csv ', usecols= lambda x:x != ' rebounds ')

#view resulting DataFrame
print (df)

  team points
0 to 22
1 B 14
2 C 29
3 D 30

Observe que a coluna bounces foi removida quando importamos o arquivo CSV para o pandas.

Se quiser remover várias colunas durante a importação, você pode usar o operador not in da seguinte maneira:

 import pandas as pd

#import all columns except 'team' and 'rebounds' into DataFrame
df=pd. read_csv (' basketball_data.csv ', usecols= lambda x:x not in [' team ', ' rebounds '])

#view resulting DataFrame
print (df)

   points
0 22
1 14
2 29
3 30

Observe que as colunas equipe e rebotes foram removidas quando importamos o arquivo CSV para o pandas.

Observe que você pode incluir quantos nomes de colunas desejar na lista após o operador not in para remover quantas colunas desejar ao importar um arquivo CSV.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras tarefas comuns em Python:

Pandas: Como pular linhas ao ler um arquivo CSV
Pandas: como adicionar dados a um arquivo CSV existente
Pandas: como usar read_csv com argumento usecols