Рабочее время между двумя сериями временных меток, исключая выходные и праздничные дни - PullRequest
0 голосов
/ 21 февраля 2019

У меня есть пандас DataFrame, который выглядит следующим образом (образец):

data = { 
    'start': ['2018-10-29 18:48:46.697000',
              '2018-10-29 19:01:10.887000',
              '2018-10-22 17:42:24.467000'], 
    'end': ['2018-10-31 17:56:38.830000',
            '2018-11-27 09:31:39.967000',
            '2018-11-28 18:33:35.243000' ]   
}
df = pd.DataFrame(data)
df['start'] = pd.to_datetime(df['start'])
df['end'] = pd.to_datetime(df['end'])

Моя цель - рассчитать (США) рабочие часы между start и end, исключая выходные и праздничные дни.Для этого я использую CustomBusinessDay функциональность pandas следующим образом:

from pandas.tseries.holiday import USFederalHolidayCalendar
from pandas.tseries.offsets import CustomBusinessDay
us_bd = CustomBusinessDay(calendar=USFederalHolidayCalendar())

len(pd.bdate_range(start=df['start'][2], end=df['end'][2], freq=us_bd))
>> 26

Это правильно с точки зрения рабочих дней (исключая выходные и праздничные дниДень благодарения и Черная пятница), но что я действительно хочу, так это количество рабочих часов между двумя временными метками.Поэтому, когда я пытаюсь использовать нативный BH:

len(pd.bdate_range(start=df['start'][2], end=df['end'][2], freq='BH'))
>> 216

, что неверно, поскольку он учитывает выходные, но не в праздничные дни.Итак, у меня есть два вопроса:

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

Когда я пытаюсь что-то вроде:

df['diff'] = pd.bdate_range(start=df['start'], end=df['end'], freq='BH')

Результат:

TypeError: Не удается преобразовать входные данные [...] типа в метку времени

Сообщение об ошибке также включает в себя весь ряд в массиве.

1 Ответ

0 голосов
/ 21 февраля 2019

Вы должны использовать CustomBusinessHour и pd.date_range вместо pd.bdate_range.

Количество часов для вашего второго ряда должно быть 145, потому что конечное время составляет 09:31:39.967.

us_bh = CustomBusinessHour(calendar=USFederalHolidayCalendar())
df['count'] = df.apply(lambda x: len(pd.date_range(start=x.start, end=x.end, freq= us_bh)),axis=1)
df['diff'] = df.apply(lambda x: pd.date_range(start=x.start, end=x.end, freq= us_bh),axis=1)
print(df)


    start                     end                  count                                               diff
0 2018-10-29 18:48:46.697 2018-10-31 17:56:38.830     16  DatetimeIndex(['2018-10-30 09:00:00', '2018-10...
1 2018-10-29 19:01:10.887 2018-11-27 09:31:39.967    145  DatetimeIndex(['2018-10-30 09:00:00', '2018-10...
2 2018-10-22 17:42:24.467 2018-11-28 18:33:35.243    200  DatetimeIndex(['2018-10-23 09:00:00', '2018-10...

И diff столбец начала рабочего часа будет '2018-10-29 09:00:00' при использовании pd.bdate_range.

us_bh = CustomBusinessHour(calendar=USFederalHolidayCalendar())
df['count'] = df.apply(lambda x: len(pd.bdate_range(start=x.start, end=x.end, freq= us_bh)),axis=1)
df['diff'] = df.apply(lambda x: pd.bdate_range(start=x.start, end=x.end, freq= us_bh),axis=1)
print(df)

                    start                     end  count                                               diff
0 2018-10-29 18:48:46.697 2018-10-31 17:56:38.830     16  DatetimeIndex(['2018-10-29 09:00:00', '2018-10...
1 2018-10-29 19:01:10.887 2018-11-27 09:31:39.967    152  DatetimeIndex(['2018-10-29 09:00:00', '2018-10...
2 2018-10-22 17:42:24.467 2018-11-28 18:33:35.243    200  DatetimeIndex(['2018-10-22 09:00:00', '2018-10...
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...