数据框转换 - Pandas

Question

我正在尝试根据工作日列的结果选择某些列。

例如：

2020 年 12 月 31 日（星期二）的站点 1 应该从 mon_open 和 mon_close 中选择值，因为它是工作日。

2020-12-29（星期日）的站点 10 应该从 sun_open 和 sun_close 中选择值，因为它是星期日。

对此高度赞赏的任何帮助。谢谢

Answer 1

您可以尝试像这样使用np.where：

df['open'] = np.where(df['weekday'].isin([5, 6]), df['sun_open'], df['mon_open'])
df['close'] = np.where(df['weekday'].isin([5, 6]), df['sun_close'], df['mon_close'])