How to update existing columns in a dataframe based on multiple conditions on other columns in python?

Question

df1:

Pan_no   Target    Debt_aum  Hml
Xxx        0        5000     Low
YYY        1          0     medium 
ZZZ        0         200     Low
Aaa        1        15000    High
Yyy        1          0      High

Condition:

If the debt_aum =0 and target =1 then hml should be Low for those rows.

Expected Output:

Pan_no   Target   Debt_aum   Hml
Xxx        0        5000     Low
YYY        1          0      Low 
ZZZ        0         200     Low
Aaa        1        15000    High
Yyy        1          0      Low

In SQL I will just write an update statement. In python I am having trouble. I tired doing

for i in df1['hml']:
   if df1[target] == 1 and df1[debt_aum] == 0:
      i = 'Low'
  else:
       i

TheFaultInOurStars · Accepted Answer · 2022-03-22 14:22:44Z

2

IIUC, you can try numpy.where function:

import pandas as pd
import numpy as np
df["Hml"] = np.where(((df["Debt_aum"] == 0) & (df["Target"])), "Low", df["Hml"])
df

Output

Pan_no	Target	Debt_aum	Hml
Xxx	0	5000	Low
YYY	1	0	Low
ZZZ	0	200	Low
Aaa	1	15000	High
Yyy	1	0	Low

answered Mar 22, 2022 at 14:22

TheFaultInOurStars

3,6331 gold badge13 silver badges30 bronze badges

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

Comments

not_speshal · Accepted Answer · 2022-03-22 14:34:42Z

2

Try with mask:

df["Hml"] = df["Hml"].mask(df["Debt_aum"].eq(0)&df["Target"].eq(1),"Low")

>>> df
  Pan_no  Target  Debt_aum   Hml
0    Xxx       0      5000   Low
1    YYY       1         0   Low
2    ZZZ       0       200   Low
3    Aaa       1     15000  High
4    Yyy       1         0   Low

answered Mar 22, 2022 at 14:34

not_speshal

23.2k2 gold badges18 silver badges33 bronze badges

2 Comments

Sasi Over a year ago

Thank you so much both the solutions worked

not_speshal Over a year ago

Happy to help. Just FYI, this is a pandas solution, no need to import numpy

SergFSM · Accepted Answer · 2022-03-22 15:26:56Z

0

a solutiont with loc:

df1.loc[(df1['Debt_aum']==0)&(df1['Target'] == 1)] = 'Low'
df1
'''
  Pan_no Target Debt_aum   Hml
0    Xxx      0     5000   Low
1    Low    Low      Low   Low
2    ZZZ      0      200   Low
3    Aaa      1    15000  High
4    Low    Low      Low   Low

answered Mar 22, 2022 at 15:26

SergFSM

1,4991 gold badge6 silver badges9 bronze badges

Collectives™ on Stack Overflow

How to update existing columns in a dataframe based on multiple conditions on other columns in python?

3 Answers 3

Output

Comments

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Output

Comments

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related