根据第二列选择列中的值

Question

我有一些DataFrame：

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

我想添加一个新列NEW，如果Day是 pizza ，则选择Food，如果Time是Food {{1}}是 pasta ，例如：

Answer 1

我们使用np.where

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

Answer 2

或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