Як видалити рядки в pandas dataframe на основі умови

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

Спосіб 1. Видалення рядків на основі однієї умови

 df = df[df. col1 > 8]

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

 df = df[(df. col1 > 8) & (df. col2 != ' A ')]

Примітка . Ми також можемо використовувати функцію drop() , щоб видаляти рядки з DataFrame, але ця функція виявилася набагато повільнішою, ніж просте призначення DataFrame відфільтрованій версії самого себе.

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

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' pos ': ['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 pos 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 на основі умови:

 #drop rows where value in 'assists' column is less than or equal to 8
df = df[df. assists > 8] 

#view updated DataFrame
df

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

Будь-який рядок із значенням, меншим або рівним 8 у стовпці «допомога», було видалено з DataFrame.

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

Наступний код показує, як видалити рядки в DataFrame на основі кількох умов:

 #only keep rows where 'assists' is greater than 8 and rebounds is greater than 5
df = df[(df. assists > 8) & (df. rebounds > 5)]

#view updated DataFrame
df

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

Єдині рядки, які ми зберігали в DataFrame, були ті, де значення передач перевищувало 8 , а значення підбирань було більше 5.

Зверніть увагу, що ми також можемо використовувати | оператор для застосування фільтра «або»:

 #only keep rows where 'assists' is greater than 8 or rebounds is greater than 10
df = df[(df. assists > 8) | (df. rebounds > 10)]

#view updated DataFrame
df

	team pos assists rebounds
0 A G 5 11
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

Єдині рядки, які ми зберігали в DataFrame, були ті, де значення допомоги було більше 8 або значення відскоку було більше 10.

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

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

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

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