使用rrule为相关月份分配天数

Question

我能够使用rrule列出两个日期之间的日期：

from dateutil import rrule
from datetime import datetime

a = '20180306'
b = '20180506'

for dt in rrule.rrule(rrule.DAILY,
                  dtstart = datetime.strptime(a, '%Y%m%d'),
                  until = datetime.strptime(b, '%Y%m%d')):
    print (dt.strftime('%Y%m%d'))

给出的输出是：

20180306
20180307
...
20180408
20180409
...
20180506

我怎样才能将这几天分配到相关月份，以便输出：

March-18: 26 days
April-18: 30 days
May-18: 6 days

我也很好奇如何扩展它以包括计算中的小时数，即。

March-18: 11.9 days

由于 的 black.mamba

Answer 1

您可以使用itertools.groupby并提取每个组的长度。在下面的示例中，grouper和res都是惰性迭代器，因此您可以通过迭代提取结果。

from dateutil import rrule
from datetime import datetime
from itertools import groupby

a = '20180306'
b = '20180506'

rule = rrule.rrule(rrule.DAILY,
                   dtstart=datetime.strptime(a, '%Y%m%d'),
                   until=datetime.strptime(b, '%Y%m%d'))

grouper = groupby(rule, key=lambda x: x.strftime('%B-%y'))

res = ((i, len(list(j))) for i, j in grouper)

for month, count in res:
    print('{0}: {1} days'.format(month, count))

March-18: 26 days
April-18: 30 days
May-18: 6 days