Преобразование данных - Панды

Question

Я пытаюсь выбрать определенные столбцы на основе результата столбца дня недели.

Например:

site1 для дня 2020-12-31 (вторник) следует выбрать значения изmon_open и mon_close, поскольку это будний день.

сайт 10 на день 2020-12-29 (воскресенье) должен выбрать значения из 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'])