[Python]：排序defaultdict的值

Question

我想按时间段对defaultdict项中的值进行排序。我如何对它们进行排序？

from collections import defaultdict
d = defaultdict(list)
d['Mon'].append("10:00-24:00")
d['Tue'].append("12:00-14:00")
d['Mon'].append("1:35-4:00")
for i in d.items():
    print i

如果我这样做，我会得到：

('Mon', ['10:00-24:00', '1:35-4:00'])
('Tue', ['12:00-14:00'])

如何按照时间顺序对defaultdict中的值进行排序，这样我就可以：

('Mon', ['1:35-4:00','10:00-24:00'])
('Tue', ['12:00-14:00'])

Answer 1

首先，你的时间应该采用一种更容易排序的格式。正如你现在所拥有的，'10:00'比'1:35'更小，这不是你想要的。解决此问题的一种方法是将前导零添加到个位数小时中的时间。

from collections import defaultdict
d = defaultdict(list)
d['Mon'].append("10:00-24:00")
d['Tue'].append("12:00-14:00")
d['Mon'].append("01:35-04:00")

for i in d.items(): i[1].sort()

>>> print d
defaultdict(<class 'list'>, {'Mon': ['01:35-04:00', '10:00-24:00'], 'Tue': ['12:00-14:00']})

如果您需要自动转换时间字符串，您可以执行regex替换：

t = '1:35-4:00'
t = re.sub(r'(^|-)(\d:\d\d)', r'\g<1>0\2', t)
>>> print t
'01:35-04:00'

还可以查看DYZ的解决方案，它可以将字符串转换为日期时间元组。他还指出，'24:00‘不是一个有效的时间，你可以照顾那些有t = t.replace('24:', '00:')的人。

Answer 2

在打印之前，按时间将值作为时间范围进行排序。下面的函数将字符串转换为两个time对象的元组：

def str2timerange(timestring):
    timestring = timestring.replace("24:00", "00:00")
    return [datetime.time(*map(int, t.split(":"))) for t in timestring.split("-")]

请记住，"24:00“不是有效时间，必须记录为"00:00”。

d = defaultdict(list)
d['Mon'].append("10:00-24:00") # WRONG, but str2timerange will fix it
d['Tue'].append("12:00-14:00")
d['Mon'].append("1:35-4:00")
for i in d.items():
    print(i[0],sorted(i[1],key=str2timerange))

#Mon ['1:35-4:00', '10:00-24:00']
#Tue ['12:00-14:00']