Conditional w/Pandas Dataframe

Question

Need a help with conditionals for pandas dataframe. Apologies in advance for the basic question or if it's covered elsewhere.

Here's the example dataframe:

employee sales revenue salary
12345    20    10000   100000

I have a few conditions based on data which will result in salary changing.

scenarios: if sales >10 and revenue > $5,000, increase salary by 20% if sales <5 and revenue > $5,000, increase salary by 10% otherwise, do nothing.

variables:
high_sales = 10
low_sales = 5
high_revenue = 5000
big_increase = 1.2
small_increase = 1.1

I know this requires some nesting but it's not clear to me how to do it.

I want the outcome to be a dataframe with only the salary column adjusted.

Here's the code:

df['salary'] = np.where((df['sales']>=high_sales & df['revenue'] 
>=high_revenue), df['salary'] * big_increase, (df['sales']<=low_sales & 
df['revenue'] >=high_revenue), df['salary'] * small_increase, df['sales'])

Is this right?

That's why I shouldn't try editing formatting from my phone. Sorry for messing it up. Please take "here's the code" out of the bottom code block to fix the string highlighting. — roganjosh
– roganjosh, Commented Sep 23, 2018 at 17:27

sacuL · Accepted Answer · 2018-09-23 17:28:01Z

1

With multiple conditions, it's nicer to use np.select rather than np.where:

conds = [(df.sales > 10) & (df.revenue > 5000),
         (df.sales < 5) & (df.revenue > 5000)]

choices = [df.salary * 1.2, df.salary * 1.1]

df['salary'] = np.select(conds, choices, default = df.revenue)

answered Sep 23, 2018 at 17:28

sacuL

51.6k9 gold badges88 silver badges115 bronze badges

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

4 Comments

jpp Over a year ago

Better, in my opinion: df['salary'] *= np.select(conds, [1.2, 1.1], 1).

sacuL Over a year ago

Yeah, perhaps that is a bit nicer, thanks for pointing it out!

nia4life Over a year ago

Thanks all. What does the "default = df.revenue" part do?

sacuL Over a year ago

That's for the otherwise, do nothing condition that you had in your post. You can take a look at the default argument in the documentation I linked

Collectives™ on Stack Overflow

Conditional w/Pandas Dataframe

1 Answer 1

4 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

4 Comments

Your Answer

Sign up or log in

Post as a guest

Related