Como excluir linhas no pandas dataframe com base na condição

Podemos usar a seguinte sintaxe para excluir linhas em um DataFrame do pandas com base na condição:

Método 1: excluir linhas com base em uma única condição

 df = df[df. col1 > 8]

Método 2: excluir linhas com base em várias condições

 df = df[(df. col1 > 8) & (df. col2 != ' A ')]

Nota : Também podemos usar a função drop() para eliminar linhas de um DataFrame, mas esta função provou ser muito mais lenta do que simplesmente atribuir o DataFrame a uma versão filtrada de si mesmo.

Os exemplos a seguir mostram como usar essa sintaxe na prática com o seguinte DataFrame do pandas:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' pos ': ['G', 'G', 'F', 'F', 'G', 'G', 'F', 'F'],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
df

	team pos assists rebounds
0 A G 5 11
1 A G 7 8
2 A F 7 10
3 A F 9 6
4 B G 12 6
5 B G 9 5
6 B F 9 9
7 B F 4 12

Método 1: excluir linhas com base em uma única condição

O código a seguir mostra como excluir linhas no DataFrame com base em uma condição:

 #drop rows where value in 'assists' column is less than or equal to 8
df = df[df. assists > 8] 

#view updated DataFrame
df

	team pos assists rebounds
3 A F 9 6
4 B G 12 6
5 B G 9 5
6 B F 9 9

Qualquer linha com valor menor ou igual a 8 na coluna “assistência” foi removida do DataFrame.

Método 2: excluir linhas com base em várias condições

O código a seguir mostra como excluir linhas no DataFrame com base em várias condições:

 #only keep rows where 'assists' is greater than 8 and rebounds is greater than 5
df = df[(df. assists > 8) & (df. rebounds > 5)]

#view updated DataFrame
df

	team pos assists rebounds
3 A F 9 6
4 B G 12 6
5 B G 9 5
6 B F 9 9

As únicas linhas que mantivemos no DataFrame foram aquelas em que o valor das assistências era maior que 8 e o valor dos rebotes era maior que 5.

Observe que também podemos usar o | operador para aplicar um filtro “ou”:

 #only keep rows where 'assists' is greater than 8 or rebounds is greater than 10
df = df[(df. assists > 8) | (df. rebounds > 10)]

#view updated DataFrame
df

	team pos assists rebounds
0 A G 5 11
3 A F 9 6
4 B G 12 6
5 B G 9 5
6 B F 9 9
7 B F 4 12

As únicas linhas que mantivemos no DataFrame foram aquelas em que o valor de assistência era maior que 8 ou o valor de recuperação era maior que 10.

Todas as linhas que não atendessem a uma dessas condições foram excluídas.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:

Como deletar linhas contendo um valor específico no Pandas
Como deletar linhas contendo uma string específica no Pandas
Como deletar linhas por índice no Pandas