Pandas: як фільтрувати рядки на основі значень у списку

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

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

У цьому конкретному прикладі DataFrame буде відфільтровано, щоб він містив лише ті рядки, де стовпець Team дорівнює значенню A , B або D.

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

Приклад: фільтруйте Pandas DataFrame на основі значень списку

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

 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

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

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

 #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

Зауважте, що відфільтрований DataFrame містить лише рядки, де значення в стовпці Team дорівнює A , B або D.

Також зауважте, що ви можете використовувати функцію isin() для фільтрації за числовими значеннями.

Наприклад, ми можемо використовувати наступний код для фільтрації рядків, у яких стовпець допомоги дорівнює 5 або 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

Зауважте, що відфільтрований DataFrame містить лише рядки зі значенням у допоміжному стовпці, що дорівнює 5 або 9 .

Примітка . Ви можете знайти повну документацію для функції pandas isin() тут .

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

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

Pandas: Як додати фільтр до зведеної таблиці
Pandas: як відфільтрувати «Не містить»
Pandas: як фільтрувати рядки, що містять певний рядок