Panda's: hoe de isin()-methode te gebruiken met de query()-methode

Vaak wilt u mogelijk de functie isin() in de methode query() in panda’s gebruiken om rijen in een DataFrame te filteren waarbij een kolom een waarde in een lijst bevat.

U kunt hiervoor de volgende syntaxis gebruiken:

 df. query (' team in ["A", "B", "D"] ')

Deze specifieke query filtert rijen uit een panda’s DataFrame waarbij de teamkolom gelijk is aan A, B of D.

Opmerking : we moeten in plaats van isin gebruiken bij het gebruik van de pandas query() -methode.

Het volgende voorbeeld laat zien hoe u deze syntaxis in de praktijk kunt gebruiken.

Voorbeeld: Gebruik de methode query() om waarden in de lijst te filteren

Stel dat we het volgende panda’s DataFrame hebben dat informatie bevat over verschillende basketbalspelers:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'B', 'B', 'C', 'C', 'D', 'E'],
                   ' 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]})

#view DataFrame
print (df)

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

Laten we nu zeggen dat we rijen willen vinden waarin de waarde in de teamkolom gelijk is aan A, B of D.

We kunnen hiervoor de volgende syntaxis gebruiken:

 #query for rows where team is in list of specific teams
df. query (' team in ["A", "B", "D"] ')

	team points assists rebounds
0 A 18 5 11
1 to 22 7 8
2 B 19 7 10
3 B 14 9 6
6 D 20 9 9

Houd er rekening mee dat de functie query() alle rijen retourneert waarin de waarde in de teamkolom gelijk is aan A, B of D.

Merk ook op dat we een lijst met teamnamen in een variabele kunnen opslaan en vervolgens naar de variabele kunnen verwijzen in de functie query() met behulp van de @ -operator:

 #create variable to hold specific team names
team_names = [" A ", " B ", " D "]

#query for rows where team is equal to a team name in team_names variable
df. query (' team in @team_names ')

	team points assists rebounds
0 A 18 5 11
1 to 22 7 8
2 B 19 7 10
3 B 14 9 6
6 D 20 9 92

De query retourneert alle rijen in het DataFrame waarin team gelijk is aan een van de teamnamen die zijn opgeslagen in de variabele team_names .

Houd er rekening mee dat de resultaten van deze zoekopdracht overeenkomen met die van het vorige voorbeeld.

Opmerking : u kunt hier de volledige documentatie van de pandas query() -functie vinden.

Aanvullende bronnen

In de volgende tutorials wordt uitgelegd hoe u andere veelvoorkomende taken in panda’s kunt uitvoeren:

Panda’s: rijen filteren op basis van tekenreekslengte
Panda’s: rijen verwijderen op basis van voorwaarde
Panda’s: hoe u het “NO IN”-filter gebruikt