búsqueda de diccionario personalizado en Python

Question

si tengo un diccionario como este

>>> d = {10: 3, 100: 2, 1000: 1} 

puedo escribir algo como:

>>> d.get(10), d.get(100), d.get(1000) 
(3, 2, 1) 

Aunque quiero que si no se encuentra la clave dada, el valor correspondiente al respeto clave más cercana se devuelve la clave dada:

>>> d.get(20), d.get(60), d.get(200) 
(3, 2, 2) 

en cambio, el resultado en Python es

(None, None, None) 

¿Cuál es una manera pitonica de implementar el comportamiento que describí?

Gracias

Answer 1

Puede derivar de dict para cambiar el comportamiento del método get():

class ClosestDict(dict): 
    def get(self, key): 
     key = min(self.iterkeys(), key=lambda x: abs(x - key)) 
     return dict.get(self, key) 

d = ClosestDict({10: 3, 100: 2, 1000: 1}) 
print (d.get(20), d.get(60), d.get(200)) 

impresiones

(3, 2, 2) 

Tenga en cuenta que la complejidad de get() ya no es O (1), pero O (norte).

Answer 2

Pedido Fuzzy matching dictionary esta receta.

Answer 3

bisect módulo permite la búsqueda rápida de la posición de inserción en una lista ordenada.

from bisect import bisect_right 

def closest_matches(data, query): 
    keys = sorted(data) 
    return [data[i] for i in (min(map(abs, (keys[p-1], keys[p]))) for p in (bisect_right(keys, k) for k in query))] 

>>> d = {10: 3, 100: 2, 1000: 1} 
>>> closest_matches(d, [20, 60, 200]) 
[3, 3, 2]