在 pandas 的查询方法中使用复杂的操作

Question

假设我有以下代码，

import pandas as pd
dat = pd.DataFrame({'col1' : [1,2,3,4,5,6,7,8,9,10], 'col2' : ['A', 'X', 'D', 'Y', 'A', 'D', 'Y', 'X', 'D', 'A']})
dat1 = pd.DataFrame({'col1' : [1,2,3,4,5,6,7,8,9,10], 'col2' : ['A', 'X', 'D', 'Y', 'A', 'D', 'Y', 'X', 'D', 'A']})
number = dat['col1'].values[dat.shape[0] - 3]
dat1.query("col1 == @number")

现在我不想在最后一行使用 @number，而是想在 query 方法中传递整个计算 dat['col1'].values[dat.shape[0] - 3]。

有什么办法可以实现吗？

Answer 1

您的具体用例是什么？ query 不支持这种复杂的表达式。

但是，您可以将 loc 与可调用项一起使用，这使您能够在管道中使用动态表达式：

dat1.loc[lambda d: d['col1'].eq(d['col1'].values[d.shape[0] - 3])]

输出：

   col1 col2
7     8    X

在 pandas 的查询方法中使用复杂的操作

Using complex operations in query method of pandas

python

pandas

python-3.6