Панды, выбирающие строки в зависимости от условия

Question

Я хотел выбрать только те строки, которые удовлетворяют моему условию.

options = ['apple', 'banana','cat','dog'] 
#selecting rows based on condition 
rslt_df = data[(data['Animal'] == "All") & data['Category'].isin(options)]
print(rslt_df)

Но я получаю эту ошибку.

       #selecting rows based on condition
-----> rslt_df = data[(data['Animal'] == "All") & data['Category'].isin(options)]
       print(rslt_df)
       TypeError: invalid type comparison

Answer 1

Эта ошибка была в некоторых старых версиях pandas, решение обновляется до последней версии, а затем использует исходное решение:

rslt_df = data[(data['Animal'] == "All") & data['Category'].isin(options)]

Answer 2

Если вы предоставите пример DataFrame, мы можем проверить, почему у вас TypeError.Но если у вас одинаковые типы данных, это должно сработать.

mask = (data['Animal'] == "All") & (data['Category'].isin(options))
new_df = df.loc[mask]