How to check if a value is in the list in selection from pandas data frame?

Question

Looks ugly:

df_cut = df_new[              (              (df_new['l_ext']==31) |              (df_new['l_ext']==22) |              (df_new['l_ext']==30) |              (df_new['l_ext']==25) |              (df_new['l_ext']==64)              )             ] 

Does not work:

df_cut = df_new[(df_new['l_ext'] in [31, 22, 30, 25, 64])] 

Is there an elegant and working solution of the above "problem"?

Answer 1

Use isin

df_new[df_new['l_ext'].isin([31, 22, 30, 25, 64])] 

Answer 2

You can use pd.DataFrame.query:

select_values = [31, 22, 30, 25, 64] df_cut = df_new.query('l_ext in @select_values') 

In the background, this uses the top-level pd.eval function.