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)