Отметить праздники в DataFrame на python3 - PullRequest
3 голосов
/ 03 ноября 2019

У меня есть пандас DataFrame с меткой времени и набором дат выходных дней

{datetime.date(2016, 1, 1), datetime.date(2016, 3, 25), datetime.date(2016, 3, 27), datetime.date(2016, 3, 28), datetime.date(2016, 5, 2), datetime.date(2016, 5, 30), datetime.date(2016, 8, 29), datetime.date(2016, 12, 25), datetime.date(2016, 12, 26), datetime.date(2016, 12, 27)}

Я хочу отметить, если это праздник или нет. Было бы здорово, если бы ты научил меня, как это сделать. Спасибо!

    timestamp
0   2016-01-01 00:00:00
1   2016-01-01 00:00:00
2   2016-01-01 00:00:00
3   2016-01-01 00:00:00
4   2016-01-01 00:00:00
... ...
20216095    2016-12-31 23:00:00
20216096    2016-12-31 23:00:00
20216097    2016-12-31 23:00:00
20216098    2016-12-31 23:00:00
20216099    2016-12-31 23:00:00

1 Ответ

1 голос
/ 03 ноября 2019

Убедитесь, что столбец timestamp имеет тип datetime, а затем используйте date для сравнения с установленными выходными.

df['timestamp'] = pd.to_datetime(df['timestamp'])
df['holidays'] = pd.np.where(df['timestamp'].dt.date.isin(holidays), 1, 0)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...