Pandas DataFrame.query()用法

为了分析数据, 我们需要大量的过滤操作。Pandas提供了query()方法来过滤DataFrame。

它提供了一种简单的选择方法, 并且还简化了基于索引的选择任务。

句法

DataFrame.query(expr, inplace=False, **kwargs)

参数

• expr：引用字符串形式的表达式以过滤数据。
• inplace：如果该值为True, 它将在原始DataFrame中进行更改。
• kwargs：引用其他关键字参数。

返回

它返回由查询表达式产生的DataFrame。

注意：仅当列名没有任何空格时, 此方法才有效。你可以将列名称中的空格替换为” _”

例1

info = pd.DataFrame({'X': range(1, 6), 'Y': range(10, 0, -2), 'Z Z': range(10, 5, -1)})
info
info.query('X > Y')
info[info.X > info.Y]
info[info.Y == info['Z Z']]

输出

X	Y			Z Z
0	1	10			10