[Python]:排序defaultdict的值

时间:2018-02-11 09:06:20

标签: python sorting defaultdict

我想根据时间段对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'])

2 个答案:

答案 0 :(得分:1)

在打印前将值按时间排序为时间范围。以下函数将您的字符串转换为两个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']

答案 1 :(得分:0)

首先,您的时间应采用更易于排序的格式。正如你所拥有的那样,'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']})

如果您需要自动转换时间字符串,可以进行正则表达式替换:

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:')的人。