数据框转换-熊猫

Question

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

例如：

2020年1月12日（星期二）的site1应该从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'])