[英]overlapping dates between two date ranges in python
我希望找到两个日期范围之间的重叠日期,如下所示:
range1 = start(2016-06-01) end (2016-06-20)
range2 = start(2016-06-10) end (2016-06-13)
结果这里是4个日期(2016-06-10,2016-06-11,2016-06-12,2016-06-13)
。 另一个例子:
range1 = start(2016-06-01) end (2016-06-20) range2 = start(2016-06-18)
end (2016-06-25)
结果这里是3个日期(2016-06-18,2016-06-19,2016-06-20)
。 如果没有日期重叠,则结果为0个日期。
我发现这篇文章有助于确定重叠日期的数量,但我想知道我是否可以在不使用长if / else语句的情况下获得实际日期?
提前致谢!
我建议在两个范围的每一个中生成日期,然后选择两个集合之间的交集。 剪切这样做可能如下所示:
from datetime import date, timedelta
def f(d1, d2):
delta = d2 - d1
return set([d1 + timedelta(days=i) for i in range(delta.days + 1)])
range1 = [date(2016, 6, 1), date(2016, 6, 20)]
range2 = [date(2016, 6, 10), date(2016, 6, 13)]
print f(*range1) & f(*range2)
出于性能目的,您还可以在生成给定范围内的日期时将d1 + timedelta(days=i)
为str
。
如果您还没有使用Python3,因此无法使用内存高效range
对象,您可以创建一个如您链接的答案中所示的命名元组(否则您也可以使用新的range
对象)。 从那里,您需要做的就是在重叠的datetime.date.fromordinal
范围内使用datetime.date.fromordinal
:
>>> from datetime import date
>>> from collections import namedtuple
>>> Range = namedtuple('Range', ['start', 'end'])
>>> r1 = Range(start=date(2016, 1, 1), end=date(2016, 2, 5))
>>> r2 = Range(start=date(2016, 1, 28), end=date(2016, 2, 28))
>>> latest_start = max(r1.start, r2.start)
>>> earliest_end = min(r1.end, r2.end)
>>> overlap = (earliest_end - latest_start).days + 1
>>> overlapping_dates = [] # default
>>> if overlap > 0:
... overlapping_dates = range(latest_start.toordinal(), earliest_end.toordinal() + 1) # as numbers
... overlapping_dates = [ date.fromordinal(x) for x in overlapping_dates ] # back to datetime.date objects
...
>>> overlapping_dates
[datetime.date(2016, 1, 28),
datetime.date(2016, 1, 29),
datetime.date(2016, 1, 30),
datetime.date(2016, 1, 31),
datetime.date(2016, 2, 1),
datetime.date(2016, 2, 2),
datetime.date(2016, 2, 3),
datetime.date(2016, 2, 4),
datetime.date(2016, 2, 5)]
使用set
的方法也会起作用(其中一种方法是在这个答案的编辑历史中),但通常效率较低,因为它必须在内存中包含所有日期,即使是那些不在交集中的日期。
我最近写了一个小函数,它返回是否有重叠然后是from-to范围
def date_overlap(start1, end1, start2, end2):
overlaps = start1 <= end2 and end1 >= start2
if not overlaps:
return False, None, None
return True, max(start1, start2), min(end1, end2)
显然,输出格式可以适应任何需要。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.