Jak zdobyć "następny" element w OrderedDict?

Question

Używam OrderedDict do losowego dostępu listę, ale teraz chcesz rzecz next na liście od tego, który mam:

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

Jak mogę dostać banana używając tylko foo i apple?

Answer 1

Jeśli są OK, z dostępu do tych części realizacji OrderedDict które są celowo utrzymywane private:

>>> 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' 

Answer 2

boję się myśleć jak wolno to będzie na liście rozmiarów, ale tylko sposób mam wymyślić tak daleko ...

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

Edit:

przykład był nieco wymyślny; moja aktualna kolekcja jest kluczowana według dat. Jeśli potrzebuję wpisu po today; znalazłem rozwiązanie przy użyciu funkcji podświetlania ...

>>> 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) 

Całkiem kęs.

Answer 3

Reworked z kodu, w ten sposób myślę, że będzie trochę lepiej:

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) 

Zasadniczo posiadające iterator na właściwym miejscu jest już wystarczająco.

Byłoby fajnie rozpocząć iterację z dowolnej pozycji, ale nie jestem pewien, czy - jak to możliwe. Potrzebuje trochę myśli.