Изменить значения столбца в Python на основе нескольких условий

Question

Я пытаюсь вставить новый столбец в Python, используя Pandas, основываясь на условии "или", но у меня проблемы с кодом. Вот что я пытаюсь сделать:

Если в столбце «Регулирующий орган» указано FDIC, ФРС или Казначейство, то я хочу, чтобы в новом столбце «Подача» было указано «Да»; в противном случае скажите «Нет». Это то, что я написал. Мой фрейм данных - df200.

    df200["Filing"] = np.where(df200["Regulatory Body"]=="FDIC","Yes","No")

Есть ли в этом коде условие "или" для соответствия двум другим переменным?

Спасибо!

Answer 1

Да. Используйте pd.Series.isin:

bodies = {'FDIC', 'Fed', 'Treasury'}

df200['Filing'] = np.where(df200['Regulatory Body'].isin(bodies), 'Yes', 'No')

В качестве альтернативы, используйте pd.Series.map с логическим массивом, полученным от pd.Series.isin:

df200['Filing'] = df200['Regulatory Body'].isin(bodies).map({True: 'Yes', False: 'No'})