pandas查询函数query的使用

query() 函数简介

pandas的query()方法是基于DataFrame列的计算代数式，对于按照某列的规则进行过滤的操作，可以使用query方法。

代码示例

import pandas as pd
df = pd.DataFrame({'a':[1, 2, 3, 4, 5, 6],
                   'b':[1, 2, 3, 4, 5, 6],
                   'c':[1, 2, 3, 4, 5, 6]})

query_list = [1, 2]
df_2 = df.query('c not in @query_list')[['a', 'b']]

使用总结

对于sql中的in或者not in，可以使用query()函数按照某列条件进行过滤，且query()函数返回一个DataFrame，可以直接在后面根据索引获取最终想要的数据。