Останній попередній робочий день у Python

Мені потрібно відняти робочі дні з поточної дати.

На даний момент у мене є якийсь код, який завжди повинен працювати в останній робочий день. Отже, це може бути сьогодні, якщо ми з понеділка по п’ятницю, але якщо це субота чи неділя, то мені потрібно повернути його на п’ятницю перед вихідними. На даний момент у мене є досить незграбний код для цього:

 lastBusDay = datetime.datetime.today()
 if datetime.date.weekday(lastBusDay) == 5:      #if it's Saturday
     lastBusDay = lastBusDay - datetime.timedelta(days = 1) #then make it Friday
 elif datetime.date.weekday(lastBusDay) == 6:      #if it's Sunday
     lastBusDay = lastBusDay - datetime.timedelta(days = 2); #then make it Friday

Чи є кращий спосіб?

Чи можу я сказати timedelta працювати в будні, а не в календарні дні, наприклад?

Використовуйте панди!

import datetime
# BDay is business day, not birthday...
from pandas.tseries.offsets import BDay

today = datetime.datetime.today()
print(today - BDay(4))

Оскільки сьогодні четвер, 26 вересня, це дасть вам результат:

datetime.datetime(2013, 9, 20, 14, 8, 4, 89761)

Здається, є кілька варіантів, якщо ви відкриті для встановлення додаткових бібліотек.

Цей пост описує спосіб визначення робочих днів за допомогою dateutil .

http://coding.derkeiler.com/Archive/Python/comp.lang.python/2004-09/3758.html

BusinessHours дозволяє вам визначити ваш перелік святкових днів тощо, щоб визначити, коли ваш робочий час (і, подовжуючи робочі дні).

http://pypi.python.org/pypi/BusinessHours/

Якщо ви хочете пропустити американські свята, а також вихідні, це спрацювало для мене (використовуючи панди 0.23.3):

import pandas as pd
from pandas.tseries.holiday import USFederalHolidayCalendar
from pandas.tseries.offsets import CustomBusinessDay
US_BUSINESS_DAY = CustomBusinessDay(calendar=USFederalHolidayCalendar())
july_5 = pd.datetime(2018, 7, 5)
result = july_5 - 2 * US_BUSINESS_DAY # 2018-7-2

Щоб перетворити на об'єкт дати python, я зробив це:

result.to_pydatetime().date()

Можливо, цей код може допомогти:

lastBusDay = datetime.datetime.today()
shift = datetime.timedelta(max(1,(lastBusDay.weekday() + 6) % 7 - 3))
lastBusDay = lastBusDay - shift

Ідея полягає в тому, що в понеділок ви повинні повернутися назад на 3 дні, у неділю 2 і 1 в будь-який інший день.

Заява (lastBusDay.weekday() + 6) % 7 просто перевизначає понеділок з 0 до 6.

Дійсно, не знаю, чи це буде краще з точки зору продуктивності.

ВІДМОВА: Я автор ...

Я написав пакет, який робить саме це, розрахунки дат бізнесу. Ви можете використовувати спеціальну специфікацію тижня та святкові дні.

У мене була саме така проблема під час роботи з фінансовими даними, і я не знайшов жодного з доступних рішень особливо легко, тому я її написав.

Сподіваюся, це корисно для інших людей.

https://pypi.python.org/pypi/business_calendar/

timeboard пакет робить це.

Припустимо, ваша дата - 04 вересня 2017 року. Незважаючи на те, що це понеділок, це було свято в США (День праці). Отже, останнім робочим днем ​​була п’ятниця, 1 вересня.

>>> import timeboard.calendars.US as US
>>> clnd = US.Weekly8x5()
>>> clnd('04 Sep 2017').rollback().to_timestamp().date()
datetime.date(2017, 9, 1)

У Великобританії 04 вересня 2017 року був звичайним робочим днем, тож останнім робочим днем ​​був сам.

>>> import timeboard.calendars.UK as UK
>>> clnd = UK.Weekly8x5()
>>> clnd('04 Sep 2017').rollback().to_timestamp().date()
datetime.date(2017, 9, 4)

ЗАМОВЛЕННЯ: Я є автором таймбордів.

Якщо хтось шукає рішення щодо свят (без величезної бібліотеки, як панди), спробуйте його функції:

import holidays
import datetime


def previous_working_day(check_day_, holidays=holidays.US()):
    offset = max(1, (check_day_.weekday() + 6) % 7 - 3)
    most_recent = check_day_ - datetime.timedelta(offset)
    if most_recent not in holidays:
        return most_recent
    else:
        return previous_working_day(most_recent, holidays)

check_day = datetime.date(2020, 12, 28)
previous_working_day(check_day)

які виробляють:

datetime.date(2020, 12, 24)

Це дасть генератор робочих днів, звичайно без свят, зупинка datetime.datetime об'єкт. Якщо вам потрібні канікули, просто наведіть додатковий аргумент із переліком канікул і уточніть у "IFology" ;-)

def workingdays(stop, start=datetime.date.today()):
    while start != stop:
        if start.weekday() < 5:
            yield start
        start += datetime.timedelta(1)

Пізніше ви зможете порахувати їх як "подобається"

workdays = workingdays(datetime.datetime(2015, 8, 8))
len(list(workdays))

Чому б вам не спробувати щось на зразок:

lastBusDay = datetime.datetime.today()
if datetime.date.weekday(lastBusDay) not in range(0,5):
    lastBusDay = 5

 def getNthBusinessDay(startDate, businessDaysInBetween):
    currentDate = startDate
    daysToAdd = businessDaysInBetween
    while daysToAdd > 0:
        currentDate += relativedelta(days=1)
        day = currentDate.weekday()
        if day < 5:
            daysToAdd -= 1

    return currentDate 

інша спрощена версія

lastBusDay = datetime.datetime.today()
wk_day = datetime.date.weekday(lastBusDay)
if wk_day > 4:      #if it's Saturday or Sunday
    lastBusDay = lastBusDay - datetime.timedelta(days = wk_day-4) #then make it Friday

Рішення незалежно від різних юрисдикцій, що мають різні свята:

Якщо вам потрібно знайти правильний ідентифікатор у таблиці, ви можете використовувати цей фрагмент. Модель таблиці - це модель sqlalchemy, і дати пошуку, за якими проводиться день поля.

def last_relevant_date(db: Session, given_date: date) -> int:
    available_days = (db.query(Table.id, Table.day)
                      .order_by(desc(Table.day))
                      .limit(100).all())
    close_dates = pd.DataFrame(available_days)
    close_dates['delta'] = close_dates['day'] - given_date
    past_dates = (close_dates
                  .loc[close_dates['delta'] < pd.Timedelta(0, unit='d')])
    table_id = int(past_dates.loc[past_dates['delta'].idxmax()]['id'])
    return table_id

Це не рішення, яке я б рекомендував, коли вам доведеться конвертувати оптом. Це досить загально і дорого, оскільки ви не використовуєте об’єднання. Більше того, передбачається, що у вас є відповідний день, який є одним із 100 останніх днів у таблиці моделей. Тому він вирішує питання введення даних, які можуть мати різні дати.