How to give multiple conditions in if elif conditions in python pandas script

Question

I have a code like:

def sample_func(new_df):

   if ( new_df['name'] == 'Tom'):
      return "Yes"
   elif( new_df['name'].isin(['Harry', 'Jerry', 'Savi', 'Aavi'])):
      return "Common Name"
   else:
      return None

I am getting error as:

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

How to fix such errors?

jezrael · Accepted Answer · 2019-07-08 09:16:16Z

3

Use numpy.select:

def sample_func(new_df):

    m1 = new_df['name'] == 'Tom'
    m2 = new_df['name'].isin(['Harry', 'Jerry', 'Savi', 'Aavi'])

    new_df['new'] = np.select([m1, m2], ['Yes','Common Name'], default=None)
    return new_df

More information about your error is here.

answered Jul 8, 2019 at 9:16

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

4 Comments

Dhiraj Over a year ago

I am getting this error now, "ValueError: shape mismatch: objects cannot be broadcast to a single shape", the scenario for which I am doing this is having different column name in variable m1 and m2.

jezrael Over a year ago

@Dhiraj - What is print (new_df.columns) ?

jezrael Over a year ago

@Dhiraj - because I guess there are duplicated column names, 2 or more name columns

Dhiraj Over a year ago

I can see the new column with the required values but can you explain me why i am getting the ValueError

Rahul charan · Accepted Answer · 2019-07-10 04:28:27Z

0

I have modified your code with any to get the result:-

def sample_func(new_df):
    if any( new_df['name'] == 'Tom'):
        return "Yes"
    elif any( new_df['name'].isin(['Harry', 'Jerry', 'Savi', 'Aavi'])):
        return "Common Name"
    else:
        return None

Output

Yes

I hope it may help you.

edited Jul 10, 2019 at 4:28

answered Jul 8, 2019 at 9:42

Rahul charan

8559 silver badges17 bronze badges

4 Comments

Ankit Agrawal Over a year ago

I don't see the reason why you converted the elif to else if nested block

Rahul charan Over a year ago

@AnkitAgrawal the first else is related to for loop AND this else contains if-else.

Ankit Agrawal Over a year ago

yes I know that but why not include it in simple if elif else block as in question @Rahulcharan

Rahul charan Over a year ago

@AnkitAgrawal Yes you can write code with if elif else also. I have modified my code, you can see it.

Collectives™ on Stack Overflow

How to give multiple conditions in if elif conditions in python pandas script

2 Answers 2

4 Comments

4 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

4 Comments

4 Comments

Your Answer

Sign up or log in

Post as a guest

Related