Change Values of Column in Python based on multiple conditions

Question

I'm trying to insert a new column in Python using Pandas based on an "or" condition, but I'm having trouble with the code. Here's what I'm trying to do:

If the column "Regulatory Body" says FDIC, Fed, or Treasury, then I want a new column "Filing" to say "Yes"; otherwise, say "No". This is what I've written. My dataframe is df200.

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

Is there a way to have an "or" condition in this code to fit the other two variables?

Thanks!

jpp · Accepted Answer · 2018-04-26 14:34:18Z

3

Yes. Use pd.Series.isin:

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

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

Alternatively, use pd.Series.map with the Boolean array you receive from pd.Series.isin:

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

edited Apr 26, 2018 at 14:34

answered Apr 26, 2018 at 14:32

jpp

166k37 gold badges301 silver badges362 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Change Values of Column in Python based on multiple conditions

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related