未来的五月一日，使得五月一日在现在和结束之间出现两次

Question

我正在运行一个优化，它应该在今天和未来的五月一日之间进行，这样五月一日会发生两次。

例如，如果 start_date = '2019-11-19' 那么我想要 end_date='2021-05-01' -> 五月一日发生两次。

如果 start_date = '2020-01-19' 那么我想要 end_date='2021-05-01' -> 五月一日发生两次。

但是当 start_date = '2020-06-19' 那么我想要 end_date='2022-05-01' -> 五月一日发生两次

如何在 Python 中做到这一点？ 基本上我想创建一个在 5 月 1 日结束的 end_date，并且还包括从今天和 end_date 开始的范围内的 5 月 1 日（包括 end_date）

Answer 1

这是我的建议：

import datetime as dt

def get_end_date(start_date):
    if start_date.month >= 5:
        end_date = dt.date(start_date.year + 2, 5, 1)
    else:
        end_date = dt.date(start_date.year + 1, 5, 1)
    return end_date

解释: 如果日期在 5 月或更晚的月份，则结束日期设置为两年后的 5 月 1 日。 否则，结束日期将设置为一年后的 5 月 1 日。 像这样，您将始终有两个 5 月 1 日（除非开始日期是 5 月 1 日，否则您将获得三个 5 月 1 日，如评论中所述）。

或者，如果您更喜欢定义 function，您可以在一行中获得相同的结果。 但是，我发现它的可读性较差： end_date = dt.date(start_date.year + 1 + int(bool(start_date.month // 5)), 5, 1)

如果您还将范围内的整个日期列表作为列表获取，您可以执行以下操作：

def get_date_range(start_date, end_date):
    delta = end_date - start_date
    return [start_date + dt.timedelta(days=i) for i in range(delta.days + 1)]

用法示例：这是您问题中给出的示例

start_date = dt.date(2019, 11, 19)
end_date = get_end_date(start_date)
print(end_date)

Output：2021-05-01

start_date = dt.date(2020, 1, 19)
end_date = get_end_date(start_date)
print(end_date)

Output：2021-05-01

start_date = dt.date(2020, 6, 19)
end_date = get_end_date(start_date)
print(end_date)

Output：2022-05-01

start_date = dt.date(2020, 5, 1)
end_date = get_end_date(start_date)
print(end_date)

Output：2022-05-01

Answer 2

好吧，结束日期与开始日期的距离永远不能超过 2 年，否则可能会超过 2 年。 所以我们可以从类似的东西开始

date_range = [start_date + timedelta(days=i) for i in range(365 * 2 + 2)]

因为 365 天 * 2 是两年。 + 2是为了说明闰年和我们从 5 月 1 日开始的额外可能性。 现在我们只需要在该列表中找到最后的 1st of may 并截断它的结尾。

from datetime import datetime, timedelta

def get_date_range(start_date):
    date_range = [start_date + timedelta(days=i) for i in range(365 * 2 + 2)]
    may_1st_indexes = [i for i, d in enumerate(date_range) if d.month == 5 and d.day == 1]
    return date_range[:may_1st_indexes[-1] + 1]

start_date = datetime.today() # This is just an example, put any other date here
result = get_date_range(start_date)

示例 output：

>>> start = datetime(2019,5,1)
>>> my_range = get_date_range(start)
>>> my_range[0]
datetime.datetime(2019, 5, 1, 0, 0)
>>> my_range[-1]
datetime.datetime(2021, 5, 1, 0, 0)