Panda's: hoe u een variabele gebruikt in de functie query().

U kunt de volgende syntaxis gebruiken om de functie query() in panda’s te gebruiken en naar een variabelenaam te verwijzen:

 df. query (' team == @team_name ')

Deze specifieke query zoekt naar rijen in een Panda DataFrame waarbij de teamkolom gelijk is aan de waarde die is opgeslagen in de variabele team_name .

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

Voorbeeld: hoe u een variabele in Panda’s-query gebruikt

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', 'A', 'B', 'B', 'B', 'B', 'C', 'C', 'C', 'C '],
                   ' position ':['G', 'G', 'F', 'G', 'F', 'F', 'F', 'G', 'G', 'F', 'F'],
                   ' points ': [22, 25, 24, 39, 34, 20, 18, 17, 20, 19, 22]})

#view DataFrame
print (df)

   team position points
0 AG 22
1 AG 25
2 AF 24
3 BG 39
4 BF 34
5 BF 20
6 BF 18
7 GC 17
8 GC 20
9 CF 19
10 CF 22

Stel nu dat we rijen willen vinden waarvan de waarde in de teamkolom gelijk is aan C.

We kunnen de volgende syntaxis gebruiken om een variabele te maken met de naam teamnaam die gelijk is aan „C“, en vervolgens naar deze variabele verwijzen in de functie query() :

 #specify team name to search for
team_name = ' C '

#query for rows where team is equal to team_name
df. query (' team == @team_name ')

        team position points
7 C G 17
8 C G 20
9 C F 19
10 C F 22

Houd er rekening mee dat de functie query() alle rijen retourneert waarin de waarde in de teamkolom gelijk is aan C.

Merk ook op dat we naar meerdere variabelen kunnen verwijzen in de functie query() als we dat willen.

De volgende code laat bijvoorbeeld zien hoe u de functie query() kunt gebruiken om alle rijen te retourneren waarin de waarde in de teamkolom gelijk is aan de waarde van een variabele met de naam team_A of een variabele met de naam team_C :

 #create two variables
team_A = ' A ' 
team_C = ' C '

#query for rows where team is equal to either of the two variables
df. query (' team == @team_A | team == @team_C ')

        team position points
0 A G 22
1 A G 25
2 A F 24
7 C G 17
8 C G 20
9 C F 19
10 C F 22

De query retourneert alle rijen in het DataFrame waarbij team gelijk is aan de waarden die zijn opgeslagen in een van de twee variabelen die we hebben opgegeven.

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