Pandy: jak filtrować wiersze na podstawie wartości na liście

Możesz użyć następującej podstawowej składni, aby filtrować wiersze w ramce DataFrame pandy, które zawierają wartość na liście:

 df[df[' team ']. isin ([' A ',' B ',' D '])]

W tym konkretnym przykładzie ramka DataFrame będzie zawierać tylko wiersze, w których kolumna Team ma wartość A , B lub D.

Poniższy przykład pokazuje, jak zastosować tę składnię w praktyce.

Przykład: filtruj ramkę danych Pandas w oparciu o wartości z listy

Załóżmy, że mamy następującą ramkę danych pand, która zawiera informacje o różnych koszykarzach:

 import pandas as pd

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

Załóżmy teraz, że chcemy przefiltrować ramkę danych tak, aby zawierała tylko wiersze, których wartość w kolumnie zespołu jest równa A , B lub D.

W tym celu możemy użyć następującej składni:

 #filter for rows where team is equal to 'A', 'B' or 'D'
df[df[' team ']. isin ([' 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
7 D 28 4 12

Należy pamiętać, że przefiltrowana ramka danych zawiera tylko wiersze, w których wartość w kolumnie Zespół jest równa A , B lub D.

Pamiętaj również, że możesz użyć funkcji isin() do filtrowania według wartości liczbowych.

Na przykład możemy użyć poniższego kodu do filtrowania wierszy, w których kolumna asyst jest równa 5 lub 9 :

 #filter for rows where assists is equal to 5 or 9
df[df[' assists ']. isin ([ 5,9 ] )]


        team points assists rebounds
0 A 18 5 11
3 B 14 9 6
5 C 11 9 5
6 D 20 9 9

Należy pamiętać, że przefiltrowana ramka danych zawiera tylko wiersze z wartością w kolumnie asyst równą 5 lub 9 .

Uwaga : pełną dokumentację funkcji panda isin() można znaleźć tutaj .

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe zadania w pandach:

Pandy: Jak dodać filtr do tabeli przestawnej
Pandy: jak filtrować „Nie zawiera”
Pandy: jak filtrować wiersze zawierające określony ciąg znaków