根据第二列选择第一列中的值

Question

我有一些数据帧：

df = pd.DataFrame({'Food': ['pizza', 'pasta', 'pizza'], 'Day': [3, 4, 5], 'Time': [8,9,10]})

我想添加一个新列NEW，如果Food是披萨的话，它选择Day，如果Food是面食的话，它选择Time，例如：

Answer 1

或loc：

df['NEW'] = df['Day']
df.loc[df['Food'].eq('pasta'), 'NEW'] = df['Time']
print(df)

输出：

   Day   Food  Time  NEW
0    3  pizza     8    3
1    4  pasta     9    9
2    5  pizza    10    5

Answer 2

我们使用np.where

df['New']=np.where(df.Food=='pizza',df.Day,df.Time)