为日期范围内的每个月生成（StartDateOfMonth，EndDateofMonth）

Question

我想为指定的日期范围生成（StartDateOfMonth，EndDateOfMonth）值的列表。例如，时间范围：2011-09-11、2013-04-24，列表应为：

[('2011-9-11', '2011-9-30'), ('2011-10-01', '2011-10-31'), ('2011-11-01', '2011-11-30'), ('2011-12-01', '2011-12-31'), ('2012-1-01', '2012-1-31'), ('2012-2-01', '2012-2-29'), ('2012-3-01', '2012-3-31'), ('2012-4-01', '2012-4-30'), ('2012-5-01', '2012-5-31'), ('2012-6-01', '2012-6-30'), ('2012-7-01', '2012-7-31'), ('2012-8-01', '2012-8-31'), ('2012-9-01', '2012-9-30'), ('2012-10-01', '2012-10-31'), ('2012-11-01', '2012-11-30'), ('2012-12-01', '2012-12-31'), ('2013-1-01', '2013-1-31'), ('2013-2-01', '2013-2-28'), ('2013-3-01', '2013-3-31'), ('2013-4-01', '2013-4-24')]

我想出了一个看起来很难看的代码。部分原因是由于我缺乏列表一致性和Python的其他功能。代码是：

def getMonthRanges(startDate, endDate):

    dateRange = []

    allYears= [eachYear for eachYear in range(startDate.year, endDate.year+1)]

    allMonths= [eachMonth for eachMonth in range(1, 13)]

    for eachYear in allYears:

        for eachMonth in allMonths:

            if eachYear == startDate.year:

                if eachMonth == startDate.month:

                   startOfMonth = str(eachYear)+'-'+str(eachMonth) + '-'+str(startDate.day)

                   endOfMonth =   str(eachYear)+ '-'+str(eachMonth) + '-'+str(calendar.monthrange(eachYear, eachMonth)[1])

                   dateRange.append((startOfMonth, endOfMonth))

                elif eachMonth > startDate.month:

                   startOfMonth = str(eachYear)+ '-'+str(eachMonth) + '-01'

                   endOfMonth = str(eachYear)+'-'+str(eachMonth)+ '-'+ str(calendar.monthrange(eachYear, eachMonth)[1])

                   dateRange.append((startOfMonth, endOfMonth))

    return dateRange

是否要求其他开发者反馈该代码是否可以压缩/改进？

Answer 1

这是仅使用datetime模块来完成此操作的一种方法：>>> from datetime import date, timedelta>>> from pprint import pprint>>> def next_month(x):&nbsp; &nbsp; &nbsp; &nbsp; 'Advance the first of the month, wrapping the year if necessary'&nbsp; &nbsp; &nbsp; &nbsp; if x.month < 12:&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; return x.replace(month=x.month+1, day=1)&nbsp; &nbsp; &nbsp; &nbsp; return x.replace(year=x.year+1, month=1)>>> def getMonthRanges(startDate, endDate):&nbsp; &nbsp; &nbsp; &nbsp; result = []&nbsp; &nbsp; &nbsp; &nbsp; first = startDate&nbsp; &nbsp; &nbsp; &nbsp; while first < endDate:&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; nm = next_month(first)&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; last = min(endDate, nm - timedelta(days=1))&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; result.append([str(first), str(last)])&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; first = nm&nbsp; &nbsp; &nbsp; &nbsp; return result>>> pprint(getMonthRanges(date(2011, 9, 11), date(2013, 4, 24)))[['2011-09-11', '2011-09-30'],&nbsp;['2011-10-01', '2011-10-31'],&nbsp;['2011-11-01', '2011-11-30'],&nbsp;['2011-12-01', '2011-12-31'],&nbsp;['2012-01-01', '2012-01-31'],&nbsp;['2012-02-01', '2012-02-29'],&nbsp;['2012-03-01', '2012-03-31'],&nbsp;['2012-04-01', '2012-04-30'],&nbsp;['2012-05-01', '2012-05-31'],&nbsp;['2012-06-01', '2012-06-30'],&nbsp;['2012-07-01', '2012-07-31'],&nbsp;['2012-08-01', '2012-08-31'],&nbsp;['2012-09-01', '2012-09-30'],&nbsp;['2012-10-01', '2012-10-31'],&nbsp;['2012-11-01', '2012-11-30'],&nbsp;['2012-12-01', '2012-12-31'],&nbsp;['2013-01-01', '2013-01-31'],&nbsp;['2013-02-01', '2013-02-28'],&nbsp;['2013-03-01', '2013-03-31'],&nbsp;['2013-04-01', '2013-04-24']]