So I want to use isin()
method with df.query()
, to select rows with id
in a list: id_list
. Similar question was asked before, but they used typical df[df['id'].isin(id_list)]
method. I'm wondering if there is a way to use df.query()
instead.
df = pd.DataFrame({'a': list('aabbccddeeff'), 'b': list('aaaabbbbcccc'),
'c': np.random.randint(5, size=12),
'd': np.random.randint(9, size=12)})
id_list = ["a", "b", "c"]
And this yields an error
df.query('a == id_list')
See Question&Answers more detail:
os 与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…