Pandas: como remover todas as colunas, exceto as específicas

Você pode usar os seguintes métodos para remover todas as colunas, exceto algumas, de um DataFrame do pandas:

Método 1: usando agulhas de crochê duplas

 df = df[[' col2 ', ' col6 ']]

Método 2: use .loc

 df = df. loc [:,[' col2 ',' col6 ']]

Ambos os métodos removem todas as colunas do DataFrame, exceto as colunas chamadas col2 e col6 .

Os exemplos a seguir mostram como usar cada método na prática com o seguinte DataFrame do pandas:

 import pandas as pd

#create DataFrame with six columns
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 28],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12],
                   ' steals ': [4, 3, 3, 2, 5, 4, 3, 8],
                   ' blocks ': [1, 0, 0, 3, 2, 2, 1, 5]})

#view DataFrame
print (df)

  team points assists rebounds steals blocks
0 A 18 5 11 4 1
1 B 22 7 8 3 0
2 C 19 7 10 3 0
3 D 14 9 6 2 3
4 E 14 12 6 5 2
5 F 11 9 5 4 2
6 G 20 9 9 3 1
7:28 4 12 8 5

Exemplo 1: Remova todas as colunas, exceto aquelas específicas, usando colchetes duplos

Podemos usar a seguinte sintaxe para remover todas as colunas do DataFrame, exceto aquelas chamadas pontos e blocos :

 #drop all columns except points and blocks
df = df[[' points ', ' blocks ']]

#view updated DataFrame
print (df)

   point blocks
0 18 1
1 22 0
2 19 0
3 14 3
4 14 2
5 11 2
6 20 1
7 28 5

Observe que apenas as colunas de pontos e blocos permanecem.

Todas as outras colunas foram removidas.

Exemplo 2: Remova todas as colunas, exceto aquelas específicas, usando .loc

Também podemos usar a função .loc para remover todas as colunas do DataFrame, exceto aquelas chamadas pontos e blocos :

 #drop all columns except points and blocks
df = df. loc [:, [' points ', ' blocks ']]

#view updated DataFrame
print (df)

   point blocks
0 18 1
1 22 0
2 19 0
3 14 3
4 14 2
5 11 2
6 20 1
7 28 5

Observe que apenas as colunas de pontos e blocos permanecem.

Isso corresponde aos resultados do exemplo anterior.

Relacionado: Pandas loc vs iloc: qual a diferença?

Recursos adicionais

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

Como remover a primeira linha no Pandas DataFrame
Como remover a primeira coluna no Pandas DataFrame
Como remover colunas duplicadas no Pandas