Pandas教程-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