如何从pandas数据框中选择一个值是否在列表中？

Question

看起来很难看:

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

不起作用:

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

是否有上述"问题"的优雅和有效解决方案？

Answer 1

使用isin

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

Answer 2

您可以使用pd.DataFrame.query：

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

在后台，这使用顶级pd.eval函数。