Using lambda if condition to column based on value of another column in Pandas dataframe

Question

Let's consider a dataframe like this:

(index)    condition    number
0          'increase'   1
1          'do-nothing' 1

I'd like to increment the elements in the number column if the condition element in the same row is equal to 'increase'. In principle, I thought about something similar:

# non-working example
df['number'] = df['number'].apply(lambda x: x+=1 if x['condition'] == 'increase' else x)

This example of course does not work. Is there an efficient method to do it?

Pierre D · Accepted Answer · 2020-12-05 17:50:35Z

1

I think you're looking for:

df.loc[df['condition'] == 'increase', 'number'] += 1

answered Dec 5, 2020 at 17:50

Pierre D

26.6k8 gold badges71 silver badges108 bronze badges

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

Collectives™ on Stack Overflow

Using lambda if condition to column based on value of another column in Pandas dataframe

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