Удаление строки DataFrame в Pandas на основе значения столбца

Если я правильно понимаю, это должно быть так же просто, как:

df = df[df.line_race != 0]

Для любых будущих пользователей, которые столкнутся с этой проблемой, стоит упомянуть, что конструкция df = df[df.line_race != 0] работает корректно, когда вы фильтруете значения, не равные 0, но не несет смысла, если вы хотите отфильтровать None или пропущенные значения.

Работает:

df = df[df.line_race != 0]

Не работает как ожидается:

df = df[df.line_race != None]

Работает:

df = df[df.line_race.notnull()]

Таким образом, для фильтрации пропущенных значений настоятельно рекомендуется использовать метод .notnull().

Чтобы добавить еще одно решение, особенно если вы используете новые ассессоры pandas, другие решения могут заменить оригинальный DataFrame и потерять ассессоры. Вот пример кода, который позволит вам удалить строки с нулевыми значениями в столбце line_race, не теряя при этом ассессоры:

df.drop(df.loc[df['line_race'] == 0].index, inplace=True)

Этот подход удаляет все строки, где значение в колонке line_race равно 0, и при этом сохраняет ваши ассессоры в DataFrame.

В случае работы с несколькими значениями в столбце с типом данных str, вы можете использовать следующую функцию для фильтрации строк:

def filter_rows_by_values(df, col, values):
    return df[~df[col].isin(values)]

В качестве примера, если у вас есть DataFrame, и вы хотите удалить строки, в которых значения "b" и "c" находятся в столбце "str", вы можете сделать это следующим образом:

import pandas as pd

df = pd.DataFrame({"str": ["a","a","a","a","b","b","c"], "other": [1,2,3,4,5,6,7]})
print(df)

# Вывод DataFrame до фильтрации
#    str  other
# 0   a      1
# 1   a      2
# 2   a      3
# 3   a      4
# 4   b      5
# 5   b      6
# 6   c      7

filtered_df = filter_rows_by_values(df, "str", ["b","c"])
print(filtered_df)

# Вывод DataFrame после фильтрации
#    str  other
# 0   a      1
# 1   a      2
# 2   a      3
# 3   a      4

Результат будет таким:

   str  other
0   a      1
1   a      2
2   a      3
3   a      4

Эта функция эффективно удаляет все строки, у которых в столбце "str" находятся запрещенные значения "b" и "c".

Если вы хотите удалить строки на основе нескольких значений столбца, вы можете использовать следующий код:

df = df[(df.line_race != 0) & (df.line_race != 10)]

Этот код удалит все строки, где значение в столбце line_race равно 0 или 10.