Создание столбца на основе If / Else со списком строковых значений для каждого условия

Question

Я пытаюсь закодировать группы фигур из набора данных о наблюдении НЛО, чтобы попытаться повысить точность модели, но я не уверен, как использовать список в качестве условия для каждого преобразования. Ниже приведен код.

def shape_group(val):
      if val == ['changed' | 'changing'| 'unknown' |'unspecified' | 'other']: # Uncertain
         return 1
      else:
          return 0

его сокращенная версия, но идея есть. Первоначально я пробовал:

 if val == ['changed' , 'changing', 'unknown' ,'unspecified' , 'other']

и

no_nan_df['shape_group'] = no_nan_df['shape'].apply(shape_type)

не возвращает совпадений

Спасибо!

Answer 1

Вы можете использовать следующее:

def shape_group(val):
      if val in ['changed', 'changing', 'unknown' , 'unspecified', 'other']: # Uncertain
         return 1
      return 0

Это будет очень быстро. Также в приведенном выше случае нет необходимости в "else".