如何根据pandas中某些列中的值从DataFrame中选择行?
在SQL中我会使用:
SELECT *
FROM table
WHERE colume_name = some_value
Run Code Online (Sandbox Code Playgroud)
我试着看看熊猫文档,但没有立即找到答案.
所以我想使用isin()方法df.query(),id在列表中选择行:id_list.之前曾问过类似的问题,但他们使用了典型的df[df['id'].isin(id_list)]方法.我想知道是否有一种方法可以使用df.query().
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"]
Run Code Online (Sandbox Code Playgroud)
这会产生错误
df.query('a == id_list')
Run Code Online (Sandbox Code Playgroud)