Jak wybierać wiersze na podstawie wielu warunków za pomocą pandas loc

Możesz użyć następujących metod, aby wybrać wiersze z ramki DataFrame pandy na podstawie kilku warunków:

Metoda 1: Wybierz wiersze spełniające wiele warunków

 df. loc [((df[' col1 '] == ' A ') & (df[' col2' ] == ' G '))]

Metoda 2: Wybierz wiersze spełniające jeden z kilku warunków

 df. loc [((df[' col1 '] > 10) | (df[' col2' ] < 8))]

Poniższe przykłady pokazują, jak w praktyce używać każdej z tych metod z następującą ramką DataFrame pand:

 import pandas as pd

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

Metoda 1: Wybierz wiersze spełniające wiele warunków

Poniższy kod pokazuje, jak wybrać tylko wiersze z ramki danych, w których zespół jest równy „A”, a pozycja jest równa „G”:

 #select rows where team is equal to 'A' and position is equal to 'G'
df. loc [((df[' team '] == ' A ') & (df[' position '] == ' G '))]

	team position assists rebounds
0 A G 5 11
1 A G 7 8

Tylko dwa wiersze w ramce danych spełniały oba te warunki.

Metoda 2: Wybierz wiersze spełniające jeden z kilku warunków

Poniższy kod pokazuje, jak wybrać tylko wiersze w ramce danych, w których asysty są większe niż 10 lub zbiórki są mniejsze niż 8:

 #select rows where assists is greater than 10 or rebounds is less than 8
df. loc [((df[' assists '] > 10) | (df[' rebounds '] < 8))]

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

Tylko trzy wiersze w ramce danych spełniały oba te warunki.

Uwaga: w tych dwóch przykładach filtrowaliśmy wiersze na podstawie dwóch warunków, ale używając & i | atrybuty . operatorów, możemy filtrować według dowolnej liczby warunków.

Dodatkowe zasoby

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

Jak utworzyć nową kolumnę na podstawie warunku w Pandach
Jak usunąć wiersze zawierające określoną wartość w Pandach
Jak usunąć zduplikowane wiersze w Pandach