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

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”?

Contents hide

Answers:

Method 1

Method 2

Answers:

Thank you for visiting the Q&A section on Magenaut. Please note that all the answers may not help you solve the issue immediately. So please treat them as advisements. If you found the post helpful (or not), leave a comment & I’ll get back to you as soon as possible.

Method 1

Use isin

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

Method 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.

All methods was sourced from stackoverflow.com or stackexchange.com, is licensed under cc by-sa 2.5, cc by-sa 3.0 and cc by-sa 4.0

0 0 votes

Article Rating