python - Set values of pandas df cell based on conditions

Question

My df is as follows:

What I want to do is, Condition: Fruit Name is NOT (Apple or Mango) and veggie Name== No Action: Set values in Veggie Color and Enjoy Eating = Unicorn

My code is

df.loc[(~df["Fruit Name"].isin(["Apple","Mango"]))& (df["Veggie Name"]=="Potato"),["Veggie Color","Enjoy Eating"]]="Unicorn"

While it does so as follows

It sets NaN to other cells

What am I missing?

Answer 1

You didn't use the exact name of the column Enjoy Eating?, so it created a new column called Enjoy Eating with NaN as default values. Just add the question mark and it will work as expected.

df.loc[(~df["Fruit Name"].isin(["Apple","Mango"]))& (df["Veggie Name"]=="Potato"),["Veggie Color","Enjoy Eating?"]]="Unicorn"