Python 将函数应用于dataframe，在参数中选择变量名称和值

我想使用一个函数，它允许我同时调用变量名（在本例中为v1）和变量值（“a”），但在这里我不能使用v1作为函数的参数

df = pd.DataFrame({'v1': pd.Categorical(["A","A","B","C"]),'v2': pd.Categorical(["N","S","W","E"])})

def select (token,x):

    comp=df[df.x==token]
    print (comp)
select("A",v1)

---

只需将按属性选择

.x

更改为更一般的

[x]

：

comp=df[df.x==token]

致：

总而言之：

def select (token,x):
    comp=df[df[x]==token]
    print (comp)
    return comp

print (select("A",'v1'))
  v1 v2
0  A  N
1  A  S

def select (token,x):
    comp=df[df[x]==token]
    print (comp)
    return comp

print (select("A",'v1'))
  v1 v2
0  A  N
1  A  S