python - Pandas 根据条件选择行

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)]