如果您只想对列表进行排序,则可以使用sorted函数,并传递key值 = 将日期字符串转换为Python的datetime对象的函数,例如lambda d: datetime.strptime(d,'%m /%Y'),请参见以下代码示例,用于您的列表
L
:
>>> from datetime import datetime
>>> sorted(L, key = lambda d: datetime.strptime(d, '%m/%Y'))
['1/2013', '2/2013', '3/2013', '7/2013', '10/2013',
'11/2013', '12/2013', '1/2014', '2/2014', '4/2014'] # indented by hand
要将“月/年字符串列表”拆分为“连续月份列表”,可以使用以下脚本(请阅读注释)。在此脚本中,我首先对列表
L
进行了排序,然后根据连续月份对字符串进行分组(为了检查连续月份,我编写了一个函数):
def is_cm(d1, d2):
""" is consecutive month pair?
: Assumption d1 is older day's date than d2
"""
d1 = datetime.strptime(d1, '%m/%Y')
d2 = datetime.strptime(d2, '%m/%Y')
y1, y2 = d1.year, d2.year
m1, m2 = d1.month, d2.month
if y1 == y2:
return (m2 - m1) == 1
elif (y2 - y1) == 1:
return (m1 == 12 and m2 == 1)
它的作用如下:
>>> is_cm('1/2012', '2/2012')
True # yes, consecutive
>>> is_cm('12/2012', '1/2013')
True # yes, consecutive
>>> is_cm('1/2015', '12/2012') # None --> # not consecutive
>>> is_cm('12/2012', '2/2013')
False # not consecutive
Code to split your code:
def result(dl):
"""
dl: dates list - a iterator of 'month/year' strings
type: list of strings
returns: list of lists of strings
"""
s_dl = sorted(dl, key=lambda d: datetime.strptime(d, '%m/%Y'))
r_dl = []
t_dl = [s_dl[0]]
for d in s_dl[1:]:
if not is_cm(t_dl[-1], d):
r_dl.append(t_dl)
t_dl = [d]
else:
t_dl.append(d)
return r_dl
result(L)
不要忘记包含
from datetime import datetime
,我相信你可以轻松地更新新日期列表,其中日期以其他格式表示。
在 @
9000 的提示下,我简化了我的排序函数并删除了旧答案,如果您想检查旧脚本,请查看 @
codepad。