尝试将秒、分钟、小时作为具有产量的可选"datetime.datetime"返回

Question

我想使用timedelta和yield，这是我的脚本：

def date_range(start_date,count_type):
time_values = {"hour": 60 * 60, "minute": 60, "second": 1}
increment_value = time_values[count_type]
while True:
yield start_date + datetime.timedelta(seconds=increment_value)
no=list(date_range(datetime.datetime(2020,4,28),"second"))
for z in no:
print(no)

当我运行脚本时，我没有从运行终端得到任何答案，你知道为什么吗？

编辑：经过一段时间的等待，我得到了这个作为答案

Traceback (most recent call last):
File "C:/U----/Desktop/countertypefunc.py", line 197, in <module>
no=list(date_range(datetime.datetime(2020,4,28),"second"))
MemoryError

Answer 1

如果我答对了问题，你想要的是一个在一定范围内生成日期(datetime对象(的生成器，并且每个连续日期之间都有一定的增量。

问题是，您通过while True创建了一个无限的generator。这意味着，如果您尝试将其强制转换为list，则会得到内存异常，因为列表不是无限的。。。

您可以做的是引入end_date(在哪里停止(，并用for循环替换while循环，该循环迭代所有应该完成的增量：

import datetime
def date_range(start_date, end_date, count_type="second"):
# determine the increment
time_values = {"hour": 60 * 60, "minute": 60, "second": 1}
incr = time_values[count_type]
# determine the number of iterations / increments to be done
n = int((end_date-start_date).total_seconds()) // incr
# iterate...
for i in range(n):
yield start_date + datetime.timedelta(seconds=incr*i)

t0, t1 = datetime.datetime(2020, 4, 28), datetime.datetime(2020, 4, 29)
for d in date_range(t0, t1, "hour"):
print(d)
# prints
# 2020-04-28 00:00:00
# 2020-04-28 01:00:00
# 2020-04-28 02:00:00
# ...
# 2020-04-28 21:00:00
# 2020-04-28 22:00:00
# 2020-04-28 23:00:00

由于使用yield，因此for循环的每次迭代将仅在调用生成器的下一个元素时执行。

您也可以使用while True，例如

def date_range2(start_date, count_type="second"):
time_values = {"hour": 60 * 60, "minute": 60, "second": 1}
i = 0
while True:
i += 1
yield start_date + datetime.timedelta(seconds=time_values[count_type]*i)
t0 = datetime.datetime(2020, 4, 28)
f = date_range2(t0)
for i in range(10):
print(next(f))
# prints
# 2020-04-28 00:00:01
# 2020-04-28 00:00:02
# 2020-04-28 00:00:03
# ...
# 2020-04-28 00:00:08
# 2020-04-28 00:00:09
# 2020-04-28 00:00:10

但使用此功能时要小心。您必须使用next()从generator获取下一个元素——如果您尝试使用list(f)，则会再次遇到while True陷阱。