Pandas: como excluir todas as linhas, exceto algumas

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

Método 1: Exclua todas as linhas, exceto aquelas com um valor específico na coluna

 #drop all rows except where team column is equal to 'Mavs'
df = df. query (" team == 'Mavs' ")

Método 2: Exclua todas as linhas, exceto aquelas que contêm um dos vários valores específicos na coluna

 #drop all rows except where team is equal to 'Mavs' or 'Heat'
df = df. query (" team == 'Mavs' | team == 'Heat' ")

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

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['Mavs', 'Mavs', 'Heat', 'Heat', 'Cavs', 'Cavs'],
                   ' points ': [18, 22, 19, 14, 14, 11],
                   ' assists ': [5, 7, 7, 9, 12, 9]})

#view DataFrame
print (df)

   team points assists
0 Mavs 18 5
1 Mavs 22 7
2 Heat 19 7
3 Heat 14 9
4 Cavs 14 12
5 Cavs 11 9

Exemplo 1: Exclua todas as linhas, exceto aquelas com um valor específico na coluna

Podemos usar a seguinte sintaxe para remover todas as linhas, exceto aquelas com o valor “Mavs” na coluna da equipe :

 #drop all rows except where team column is equal to 'Mavs'
df = df. query (" team == 'Mavs' ")

#view updated DataFrame
print (df)

   team points assists
0 Mavs 18 5
1 Mavs 22 7

Observe que todas as linhas foram removidas, exceto as linhas que possuem o valor “Mavs” na coluna da equipe .

Exemplo 2: Exclua todas as linhas, exceto aquelas que contêm um dos vários valores específicos na coluna

Podemos usar a seguinte sintaxe para remover todas as linhas, exceto aquelas com valor “Mavs” ou “Heat” na coluna da equipe :

 #drop all rows except where team column is equal to 'Mavs'
df = df. query (" team == 'Mavs' | team == 'Heat' ")

#view updated DataFrame
print (df)

   team points assists
0 Mavs 18 5
1 Mavs 22 7
2 Heat 19 7
3 Heat 14 9

Observe que todas as linhas foram removidas, exceto as linhas que possuem o valor “Mavs” ou “Heat” na coluna Equipe .

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