如何在UDF函数中使用Isin？

Question

我正在尝试使用带有isin的UDF函数根据数据框中其他变量的值创建一个新变量。

df = pd.DataFrame({'A': [1, 2, 3,4,5,6], 'B': ['a', 'b', 'f']*2})

def fun(c):
    if c.isin([1, 3]):
        val = 'yes'
    else:
        val = 'no'
    return val


df['pr']=df['A'].apply(fun)

结果是此错误： AttributeError：“ int”对象没有属性“ isin”