pandas-query使用

Posted 2023-03-14

参考技术A 上午发现了一个很好用的函数，query，这里详细记录学习下

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

具体的使用方法，可以参考官方文档，这里我记录些我觉得好玩儿的内容
官方介绍：

使用query函数，可以通过 @ 引用当前的变量
数据集的话，使用的是上一篇的内容： pandas实例-筛选与过滤-Fictional Army

注意几个点：

这里直接使用index代表了

就和SQL里一样使用的，看上面的那个例子

通常 == 和 ！= 表示相等或不相等，但是在query中，可以用来代替 in

query中同样适用Boolean，就和布尔数组一样的玩儿法，但又稍微有点儿特殊

这里把布尔值作为了一列，要不然query中是找不到的，

这里还可以配合 not 或者 ~ 一起使用表示取反，就是非的意思

使用下来，感觉query，就是和SQL一样的，很好用