我有一些DataFrame:
df = pd.DataFrame({'Food': ['pizza', 'pasta', 'pizza'], 'Day': [3, 4, 5], 'Time': [8,9,10]})
我想添加一个新列,NEW,如果Day为pizza,则选择Food;如果Time为pasta,则选择Food,例如:
NEW
Day
Food
Time
我们使用np.where
np.where
df['New']=np.where(df.Food=='pizza',df.Day,df.Time)