Как получить "следующий" элемент в OrderedDict?

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

foo = OrderedDict([('apple', 4), ('banana', 3), ('orange', 2), ('pear', 1)])
apple = foo['apple']

Как мне получить банан, используя только foo и apple?

Ответ 1

Если вы в порядке с доступом к тем частям реализации OrderedDict, которые намеренно сохраняются конфиденциальными:

>>> class MyOrderedDict(OrderedDict):
...     def next_key(self, key):
...             next = self._OrderedDict__map[key][1]
...             if next is self._OrderedDict__root:
...                     raise ValueError("{!r} is the last key".format(key))
...             return next[2]
...     def first_key(self):
...             for key in self: return key
...             raise ValueError("OrderedDict() is empty")
... 
>>> od = MyOrderedDict([('apple', 4), ('banana', 3), ('orange', 2), ('pear', 1)])
>>> od.next_key("apple")
'banana'
>>> od.next_key("banana")
'orange'
>>> od.next_key("orange")
'pear'
>>> od.next_key("pear")
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 5, in next_key
ValueError: 'pear' is the last key
>>> od.first_key()
'apple'

Ответ 2

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

>>> foo.items()[foo.keys().index('apple') + 1]
('banana', 3)

Edit:

Пример был слегка надуман; моя фактическая коллекция зависит от дат. Если мне нужна запись после today; нашел решение, использующее dropwhile...

>>> foo = OrderedDict([(datetime.date(2000,1,1), 4), (datetime.date(2000,5,23), 3), datetime.date(2000,10,1), 2), (datetime.date(2000,12,31), 1)])
>>> today = datetime.date(2000,1,30)
>>> foo.items()[foo.keys().index((itertools.dropwhile(lambda d: d<today, foo)).next())]
(datetime.date(2000, 5, 23), 3)

Довольно глоток.

Ответ 3

Переработанный из вашего кода, таким образом, я думаю, будет немного лучше:

import collections as co
import datetime as dt
import itertools as it

foo = co.OrderedDict([
    (dt.date(2000,1,1), 4),
    (dt.date(2000,5,23), 3),
    (dt.date(2000,10,1), 2),
    (dt.date(2000,12,31), 1)
])
today = dt.date(2000,1,30)

fooiter = it.dropwhile(lambda d: d <= today, foo)
print next(fooiter)
print list(fooiter)

В основном наличие итератора в нужном месте уже достаточно.

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

Ответ 4

Python 3.X

dict.items будет возвращать итеративный объект представления dict, а не список. Нам нужно заключить вызов в список, чтобы сделать возможной индексацию:

>>> from collections import OrderedDict
>>> 
>>> foo = OrderedDict([('apple', 4), ('banana', 3), ('orange', 2), ('pear', 1)])
>>> 
>>> def next_item(odic, key):
...     return list(odic)[list(odic.keys()).index(key) + 1]
... 
>>> next = next_item(foo, 'apple')
>>> print(next, foo[next])
banana 3