Pandas: як використовувати метод isin() із методом query().

Часто ви можете використовувати функцію isin() у методі query() у pandas, щоб фільтрувати рядки у DataFrame, де стовпець містить значення у списку.

Для цього можна використати такий синтаксис:

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

Цей конкретний запит фільтрує рядки з pandas DataFrame, де командний стовпець дорівнює A, B або D.

Примітка : нам потрібно використовувати in замість isin під час використання методу pandas query() .

У наступному прикладі показано, як використовувати цей синтаксис на практиці.

Приклад: використовуйте метод query() для фільтрації значень у списку

Припустімо, що у нас є такий фрейм даних pandas, який містить інформацію про різних баскетболістів:

 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

Тепер припустімо, що ми хочемо знайти рядки, де значення в стовпці team дорівнює A, B або D.

Для цього ми можемо використати такий синтаксис:

 #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

Зауважте, що функція query() повертає всі рядки, де значення в стовпці team дорівнює A, B або D.

Також зауважте, що ми можемо зберегти список імен команд у змінній, а потім посилатися на змінну у функції query() за допомогою оператора @ :

 #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

Запит повертає всі рядки у DataFrame, де team дорівнює одному з імен команд, збережених у змінній team_names .

Зауважте, що результати цього запиту збігаються з результатами попереднього прикладу.

Примітка . Повну документацію щодо функції pandas query() можна знайти тут .

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

У наступних посібниках пояснюється, як виконувати інші типові завдання в pandas:

Pandas: як фільтрувати рядки на основі довжини рядка
Pandas: як видалити рядки на основі умови
Pandas: як використовувати фільтр «NO IN».