web-dev-qa-db-ja.com

値がpandasデータフレームから選択したリストにあるかどうかを確認する方法

見苦しい:

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

上記の「問題」のエレガントで実用的なソリューションはありますか?

23
Roman

isin を使用します

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

pd.DataFrame.query

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

バックグラウンドでは、これはトップレベル pd.eval 関数。

0
jpp