如果在列表函数中，则为 python 语法

Question

如果 columnx 包含以下任何数字，我正在尝试找到正确的 python 语法来创建值为 yes 的标志: 1、2、3、4、5。

def create_flag(df):
    if df['columnx'] in (1,2,3,4,5):
        return df['flag']=='yes'

我收到以下错误。

TypeError: invalid type comparison

我的语法有明显错误吗？

Answer 1

将 np.where 与 pandas isin 一起使用:

df['flag'] = np.where(df['columnx'].isin([1,2,3,4,5]),'yes','no')