Ngày làm việc bằng Python

Question

Tôi cần trừ ngày làm việc từ ngày hiện tại.

Tôi hiện có một số mã cần phải luôn chạy trong ngày làm việc gần đây nhất. Vì vậy, đó có thể là ngày hôm nay nếu chúng tôi là thứ hai đến thứ sáu, nhưng nếu đó là thứ bảy hoặc chủ nhật sau đó tôi cần phải thiết lập nó trở lại thứ sáu trước khi cuối tuần. Tôi hiện đang có một số khá clunky mã để làm điều này:

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 

Có cách nào tốt hơn?

Tôi có thể yêu cầu timedelta hoạt động trong các ngày trong tuần thay vì ngày theo lịch không?

Answer 1

Sử dụng gấu trúc!

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

# pd.datetime is an alias for datetime.datetime 
today = pd.datetime.today() 
print today - BDay(4) 

Kể từ hôm nay là Thứ Năm Tháng 9 26, mà sẽ cung cấp cho bạn một sản lượng:

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

Answer 2

Dường như có một số tùy chọn nếu bạn mở để cài đặt thêm thư viện.

Bài đăng này mô tả cách xác định ngày làm việc với dateutil.

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

BusinessHours cho phép bạn tùy chỉnh xác định danh sách các ngày lễ, vv, để xác định khi nào giờ bạn làm việc (và bằng cách mở rộng ngày làm việc) đang có.

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

Answer 3

Có lẽ mã này có thể giúp:

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

Ý tưởng là vào thứ Hai yo phải quay trở lại 3 ngày, vào ngày chủ nhật 2, và 1 trong bất kỳ ngày nào khác.

Tuyên bố (lastBusDay.weekday() + 6) % 7 chỉ cần tái căn cứ thứ hai từ 0 đến 6.

Thực sự không biết nếu điều này sẽ tốt hơn về hiệu suất.

Answer 4

Tại sao bạn không thử một cái gì đó như:

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

Answer 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 

Answer 6

DISCLAMER: Tôi là tác giả ...

tôi đã viết một gói thực hiện chính xác điều này, tính toán ngày làm việc. Bạn có thể sử dụng thông số và ngày lễ của tuần tùy chỉnh.

Tôi đã gặp sự cố chính xác này khi làm việc với dữ liệu tài chính và không tìm thấy bất kỳ giải pháp có sẵn nào dễ dàng, vì vậy tôi đã viết một giải pháp.

Hy vọng điều này hữu ích cho người khác.

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

Answer 7

này sẽ cung cấp một máy phát điện của ngày làm việc, tất nhiên không có ngày nghỉ, dừng là datetime.datetime đối tượng. Nếu bạn cần ngày lễ chỉ làm cho lập luận thêm với danh sách các ngày lễ và kiểm tra với 'IFology' ;-)

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

Sau đó bạn có thể đếm chúng như

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