在Python中遍历日期范围
我有以下代码来实现这个功能,但我想知道怎么做得更好?现在我觉得这比嵌套循环要好,但当你在列表推导式中使用生成器时,它开始变得像Perl的一行代码那样复杂。
day_count = (end_date - start_date).days + 1
for single_date in [d for d in (start_date + timedelta(n) for n in range(day_count)) if d <= end_date]:
print strftime("%Y-%m-%d", single_date.timetuple())
注意事项
- 我实际上并不是用这个来打印。那只是为了演示。
start_date
和end_date
变量是datetime.date
对象,因为我不需要时间戳。(它们将用于生成报告)。
示例输出
对于开始日期2009-05-30
和结束日期2009-06-09
:
2009-05-30
2009-05-31
2009-06-01
2009-06-02
2009-06-03
2009-06-04
2009-06-05
2009-06-06
2009-06-07
2009-06-08
2009-06-09
28 个回答
206
使用 dateutil
这个库:
from datetime import date
from dateutil.rrule import rrule, DAILY
a = date(2009, 5, 30)
b = date(2009, 6, 9)
for dt in rrule(DAILY, dtstart=a, until=b):
print dt.strftime("%Y-%m-%d")
这个Python库有很多高级功能,其中一些非常实用,比如 relative delta
(相对增量)——而且它是一个单独的文件(模块),很容易就能加入到你的项目中。
366
这可能会更清楚一些:
from datetime import date, timedelta
start_date = date(2019, 1, 1)
end_date = date(2020, 1, 1)
delta = timedelta(days=1)
while start_date <= end_date:
print(start_date.strftime("%Y-%m-%d"))
start_date += delta
801
为什么会有两个嵌套的循环?对我来说,只用一个循环就能产生相同的数据列表:
for single_date in (start_date + timedelta(n) for n in range(day_count)):
print ...
而且没有列表被存储,只有一个生成器在被迭代。而且生成器里的“if”似乎也是多余的。
毕竟,一个线性序列只需要一个迭代器,而不是两个。
与John Machin讨论后的更新:
也许最优雅的解决方案是使用生成器函数,完全隐藏或抽象掉对日期范围的迭代:
from datetime import date, timedelta
def daterange(start_date, end_date):
for n in range(int((end_date - start_date).days)):
yield start_date + timedelta(n)
start_date = date(2013, 1, 1)
end_date = date(2015, 6, 2)
for single_date in daterange(start_date, end_date):
print(single_date.strftime("%Y-%m-%d"))
注意:为了与内置的 range()
函数保持一致,这个迭代在到达 end_date
之前就停止了。所以如果想要包含最后一天,就要用第二天,就像使用 range()
一样。