python - 一周结束日期，周日为一周的第一天

Question

我正在尝试获取日期列表的周结束日期。我找到了下面的代码，它适用于传统的周一到周日，但我需要周日到周六(周日是一周的第一天)。

from datetime import datetime, timedelta

date_str = '2019-06-16'
date_obj = datetime.strptime(date_str, '%Y-%m-%d')

start_of_week = date_obj - timedelta(days=date_obj.weekday())
end_of_week = start_of_week + timedelta(days=6)  

print(start_of_week)
print(end_of_week)

上面的代码返回:

2019-06-10 00:00:00

2019-06-16 00:00:00

我需要它返回这个:

2019-06-16 00:00:00

2019-06-22 00:00:00

Answer 1

因此，工作日返回 0 表示星期一，6 表示星期日。要将周日更改为 0，周六更改为 6，请执行

start_of_week = date_obj - timedelta((date_obj.weekday() + 1) % 7)