python - 从字典到列表的日期

Question

这对我来说太复杂了，所以我正在寻求帮助。我有这样的命令:

dict = {'date_from': u'2019-04-01', 
        'date_to': u'2019-04-30'}

根据这个字典，我需要列一个这样的列表:

list = ['2019-04']

因为月份相同。

如果dict是这样的

dict = {'date_from': u'2019-04-01', 
        'date_to': u'2019-05-30'}

列表应该是这样的:

list = ['2019-04', '2019-05']

等等。

另一个例子

dict = {'date_from': u'2019-01-01', 
         'date_to': u'2019-05-30'}


list = ['2019-01', '2019-02','2019-03','2019-04','2019-05']

我知道这是可能的，但不知道如何让它发挥作用。

Answer 1

使用datetime & relativedelta

例如:

import datetime
from dateutil.relativedelta import relativedelta

d = {'date_from': u'2019-01-01', 'date_to': u'2019-05-30'}
result = []
start = datetime.datetime.strptime(d["date_from"], "%Y-%m-%d") 
end = datetime.datetime.strptime(d["date_to"], "%Y-%m-%d")
while True:
    if start > end:
        break
    result.append(start.strftime("%Y-%m"))
    start = start + relativedelta(months=1)
print(result)   

输出:

['2019-01', '2019-02', '2019-03', '2019-04', '2019-05']