Як вибрати рядки на основі кількох умов за допомогою pandas loc

Ви можете використовувати такі методи, щоб вибрати рядки з pandas DataFrame на основі кількох умов:

Спосіб 1. Виберіть рядки, які відповідають кільком умовам

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

Спосіб 2. Виберіть рядки, які відповідають одній із кількох умов

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

Наступні приклади показують, як використовувати кожен із цих методів на практиці з такими pandas DataFrame:

 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

Спосіб 1. Виберіть рядки, які відповідають кільком умовам

Наступний код показує, як вибрати лише рядки з DataFrame, де команда дорівнює “A”, а позиція дорівнює “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

У DataFrame було лише два рядки, які відповідали обом цим умовам.

Спосіб 2. Виберіть рядки, які відповідають одній із кількох умов

Наступний код показує, як вибрати лише ті рядки в DataFrame, у яких кількість передач перевищує 10 або кількість підбирань менше 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

У DataFrame було лише три рядки, які відповідали обом цим умовам.

Примітка. У цих двох прикладах ми відфільтрували рядки на основі двох умов, але використовуючи & та | атрибути . операторів, ми можемо фільтрувати скільки завгодно умов.

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції в pandas:

Як створити новий стовпець на основі умови в Pandas
Як видалити рядки, що містять певне значення в Pandas
Як видалити повторювані рядки в Pandas