В Python, как перебрать словарь в порядке сортировки ключей? - PullRequest
197 голосов
/ 13 декабря 2008

Существует существующая функция, которая заканчивается следующим: d - словарь:

return d.iteritems()

, который возвращает несортированный итератор для данного словаря. Я хотел бы вернуть итератор, который просматривает элементы, отсортированные по ключу . Как мне это сделать?

Ответы [ 10 ]

160 голосов
/ 13 декабря 2008

Не тестировал это очень подробно, но работает в Python 2.5.2.

>>> d = {"x":2, "h":15, "a":2222}
>>> it = iter(sorted(d.iteritems()))
>>> it.next()
('a', 2222)
>>> it.next()
('h', 15)
>>> it.next()
('x', 2)
>>>

Если вы привыкли делать for key, value in d.iteritems(): ... вместо итераторов, это все равно будет работать с решением выше

>>> d = {"x":2, "h":15, "a":2222}
>>> for key, value in sorted(d.iteritems()):
>>>     print(key, value)
('a', 2222)
('h', 15)
('x', 2)
>>>

В Python 3.x используйте d.items() вместо d.iteritems() для возврата итератора.

80 голосов
/ 13 декабря 2008

Используйте функцию sorted():

return sorted(dict.iteritems())

Если вам нужен реальный итератор для отсортированных результатов, поскольку sorted() возвращает список, используйте:

return iter(sorted(dict.iteritems()))
38 голосов
/ 13 декабря 2008

Ключи диктанта хранятся в хеш-таблице, так что это их «естественный порядок», то есть псевдослучайный. Любой другой заказ - это понятие потребителя диктата.

sorted () всегда возвращает список, а не диктат. Если вы передадите ему dict.items () (который создает список кортежей), он вернет список кортежей [(k1, v1), (k2, v2), ...], которые можно использовать в цикле в некотором смысле очень похоже на диктовку, но это ни в коем случае не диктат !

foo = {
    'a':    1,
    'b':    2,
    'c':    3,
    }

print foo
>>> {'a': 1, 'c': 3, 'b': 2}

print foo.items()
>>> [('a', 1), ('c', 3), ('b', 2)]

print sorted(foo.items())
>>> [('a', 1), ('b', 2), ('c', 3)]

Следующее выглядит как дикт в цикле, но это не так, это список кортежей, распаковываемых в k, v:

for k,v in sorted(foo.items()):
    print k, v

Примерно эквивалентно:

for k in sorted(foo.keys()):
    print k, foo[k]
30 голосов
/ 13 декабря 2008

Грег ответил правильно. Обратите внимание, что в Python 3.0 вам придется делать

sorted(dict.items())

как iteritems исчезнет.

6 голосов
/ 08 июля 2013

Теперь вы можете использовать OrderedDict в Python 2.7:

>>> from collections import OrderedDict
>>> d = OrderedDict([('first', 1),
...                  ('second', 2),
...                  ('third', 3)])
>>> d.items()
[('first', 1), ('second', 2), ('third', 3)]

Здесь у вас есть что нового страница для версии 2.7 и OrderedDict API .

5 голосов
/ 07 марта 2013

В общем, сортировать диктовку можно так:

for k in sorted(d):
    print k, d[k]

Для конкретного случая в вопросе, имеющего «замену» для d.iteritems (), добавьте функцию, подобную:

def sortdict(d, **opts):
    # **opts so any currently supported sorted() options can be passed
    for k in sorted(d, **opts):
        yield k, d[k]

и поэтому конечная строка меняется с

return dict.iteritems()

до

return sortdict(dict)

или

return sortdict(dict, reverse = True)
4 голосов
/ 11 апреля 2013
>>> 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), однако после короткого линейного heapify он возвращает элементы в отсортированном порядке, что делает его теоретически более эффективным, когда не всегда нужен весь список.

3 голосов
/ 09 декабря 2011

Если вы хотите отсортировать по порядку, в который элементы были вставлены, а не по порядку ключей, вам следует взглянуть на collection в Python .OrderedDict . (Только Python 3)

3 голосов
/ 13 декабря 2008

sorted возвращает список, следовательно, ваша ошибка при попытке перебрать его, но поскольку вы не можете заказать диктовку, вам придется иметь дело со списком.

Я понятия не имею, каков более широкий контекст вашего кода, но вы можете попробовать добавить итератор в результирующий список. как это может быть?:

return iter(sorted(dict.iteritems()))

конечно, теперь вы будете возвращать кортежи, потому что сортировка превратила ваш диктант в список кортежей

например: скажи, что твой голос был: {'a':1,'c':3,'b':2} отсортированный превращает его в список:

[('a',1),('b',2),('c',3)]

поэтому, когда вы на самом деле перебираете список, вы получаете (в этом примере) кортеж состоит из строки и целого числа, но, по крайней мере, вы сможете перебирать его.

2 голосов
/ 18 июня 2012

Если вы используете CPython 2.x и имеете большой словарь mydict, то использование sorted (mydict) будет медленным, потому что sorted создает отсортированный список ключей mydict.

В этом случае вы, возможно, захотите взглянуть на мой пакет orderdict, который включает реализацию C на sorteddict в C. Особенно, если вам приходится просматривать отсортированный список ключей несколько раз на разных этапах (т. Е. Количество элементов ) время жизни словарей.

http://anthon.home.xs4all.nl/Python/ordereddict/

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...