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”?
Answers:
Use isin
df_new[df_new['l_ext'].isin([31, 22, 30, 25, 64])]
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.
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”?
Use isin
df_new[df_new['l_ext'].isin([31, 22, 30, 25, 64])]
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.