Pandas：如何应用传递的条件运算符来选择pandas中的行？

Question

如何将操作符作为参数传递给函数，并在该函数内部使用该运算符来选择pandas行？

# helper.py
def get_milage(operator, condition):
    return df[df.Milage <operator> condition]

# main.py
import operator
get_milage(operator.lt, 7)

Answer 1

我认为需要在()中使用operator，与operator.lt(column, value)相同：

# helper.py
def get_milage(operator, condition):
    return df[operator(df.Milage, condition)]