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'

Answer 1

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

Answer 2

You need:

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