有一个现有的函数以下面的形式结束,其中d是一个字典:
return d.iteritems()
它返回给定字典的未排序迭代器。我想返回一个迭代器,遍历按键排序的项。我怎么做呢?
有一个现有的函数以下面的形式结束,其中d是一个字典:
return d.iteritems()
它返回给定字典的未排序迭代器。我想返回一个迭代器,遍历按键排序的项。我怎么做呢?
当前回答
>>> import heapq
>>> d = {"c": 2, "b": 9, "a": 4, "d": 8}
>>> def iter_sorted(d):
keys = list(d)
heapq.heapify(keys) # Transforms to heap in O(N) time
while keys:
k = heapq.heappop(keys) # takes O(log n) time
yield (k, d[k])
>>> i = iter_sorted(d)
>>> for x in i:
print x
('a', 4)
('b', 9)
('c', 2)
('d', 8)
这个方法仍然有一个O(N log N)排序,然而,在一个简短的线性堆化之后,它会生成有序的条目,当你不总是需要整个列表时,理论上它更有效。
其他回答
你现在也可以在Python 2.7中使用OrderedDict:
>>> from collections import OrderedDict
>>> d = OrderedDict([('first', 1),
... ('second', 2),
... ('third', 3)])
>>> d.items()
[('first', 1), ('second', 2), ('third', 3)]
这里有2.7版本的what's new页面和OrderedDict API。
>>> import heapq
>>> d = {"c": 2, "b": 9, "a": 4, "d": 8}
>>> def iter_sorted(d):
keys = list(d)
heapq.heapify(keys) # Transforms to heap in O(N) time
while keys:
k = heapq.heappop(keys) # takes O(log n) time
yield (k, d[k])
>>> i = iter_sorted(d)
>>> for x in i:
print x
('a', 4)
('b', 9)
('c', 2)
('d', 8)
这个方法仍然有一个O(N log N)排序,然而,在一个简短的线性堆化之后,它会生成有序的条目,当你不总是需要整个列表时,理论上它更有效。
使用sorted()函数:
return sorted(dict.iteritems())
如果你想在排序结果上使用一个实际的迭代器,因为sorted()返回一个列表,请使用:
return iter(sorted(dict.iteritems()))
格雷格的答案是正确的。注意,在Python 3.0中,你必须这样做
sorted(dict.items())
因为iteritems会消失。
如果你想按插入项的顺序排序,而不是按键的顺序排序,你应该看看Python的collections.OrderedDict。(仅限Python 3)