Pandas:获取与查询关联的掩码

2024-06-07 06:08:33 发布

您现在位置:Python中文网/ 问答频道 /正文

我喜欢熊猫中的query方法。与常规的选择方法相比,它速度快,表现力强,而且不太冗长。在

给定一个查询,是否可以得到True/False掩码,它对应于query返回的值?在

例如,假设我有:

my_query = 'values >= {0} and values <= {1}'.format(Q1, Q2)
inliers  = df.query(my_query)

inliers将保存满足查询的数据,但我是否也可以获得此查询的掩码?在

获取掩码可能很有用,例如,快速否定查询,或获得与原始数据帧大小相同的结果。在


Tags: and方法falsetrueformatmyquery速度
1条回答
网友
1楼 · 发布于 2024-06-07 06:08:33

使用df.eval()df.query()基本上是{}

In [32]: df = DataFrame(dict(A = range(5)))

In [33]: df
Out[33]: 
   A
0  0
1  1
2  2
3  3
4  4

In [34]: df.query('A>3')
Out[34]: 
   A
4  4

In [36]: df.eval('A>3')
Out[36]: 
0    False
1    False
2    False
3    False
4     True
dtype: bool

相关问题 更多 >