Выберите выходные и рабочие дни по дням недели () в python

Question

У меня есть следующие данные:

2010-01-01 00:00:00 327
2010-01-01 01:00:00 237
2010-01-01 02:00:00 254
...

Я могу выбирать разные данные по месяцам, но я хотел бы выбрать данные по будням и данные по выходным. Как это можно было сделать, используя функцию буднего дня? Я проделал что-то подобное, но это не сработало.

data['Month'] = data['Timestamp'].dt.month
summer = (data.Month >=6) & (data.Month <=8) 
data['WEEKDAY'] = ((pd.DatetimeIndex(data.index).dayofweek) // 5 == 1).astype(float)

Answer 1

Используйте:

data['WEEKDAY'] = data.Timestamp.dt.dayofweek

0 - понедельник. Затем отметьте выходные как 1 в столбце WEEKEND:

data['WEEKEND'] = np.where(data.Timestamp.dt.dayofweek.isin([5,6]), 1, 0)

Чтобы разделить на 2 фрейма данных:

weekends = data[data['WEEKEND']==1]
workdays = data[data['WEEKEND']!=1]