如何使用df.函数中的查询方法?

Question

我正在尝试创建一个函数，该函数将使用查询方法根据多个条件过滤数据帧，并创建一个具有适用值的新系列。

下面是我正在使用的df的一个例子:

100 150149350500600101130147

一个可能的解决方案，基于f strings:

def sorter(df,week):
return df.query(f'Week == {week} & Treatment.str.startswith("A").values')['Value']

作为另一种选择，您可以使用@定义变量:

def sorter(df,week):
return df.query('Week == @week & Treatment.str.startswith("A").values')['value']