check and update column based on multiple values

Question

I have a dataframe df where there are 3 columns

id code  status   
1  US    Y      
2  IN    Y
3  UK    Y
4  CN    Y
5  KR    Y

I want to update column status to N where code not in ("US", "UK")

I tried using this but failed

df.loc[df['code'] not in ("US","UK"),["status"]] ='N'

Possible duplicate of [Pandas: How do I assign values based on multiple conditions for existing columns?](https://stackoverflow.com/questions/30631841/pandas-how-do-i-assign-values-based-on-multiple-conditions-for-existing-columns) — Mohit Motwani, Aug 10 '18 at 06:49

score 1 · Accepted Answer · answered Aug 10 '18 at 06:45

1

You need:

df['status'] = np.where(df['code'].isin(["US","UK"]), df['status'], 'N')

answered Aug 10 '18 at 06:45

Sociopath

score 1 · Answer 2 · answered Aug 10 '18 at 06:49

1

df['status'] = df.apply(lambda x: 'N' if x[1] not in ['US','UK'] else x[2],axis=1)

answered Aug 10 '18 at 06:49

Onyambu

2 Answers2