¿Cómo implementar un dictado predeterminado ordenado?
Me gustaría combinar OrderedDict()y defaultdict()from collectionsen un objeto, que será un valor predeterminado ordenado dict.
es posible?
Lo siguiente (usando una versión modificada de esta receta ) me funciona:
from collections import OrderedDict, Callable
class DefaultOrderedDict(OrderedDict):
# Source: http://stackoverflow.com/a/6190500/562769
def __init__(self, default_factory=None, *a, **kw):
if (default_factory is not None and
not isinstance(default_factory, Callable)):
raise TypeError('first argument must be callable')
OrderedDict.__init__(self, *a, **kw)
self.default_factory = default_factory
def __getitem__(self, key):
try:
return OrderedDict.__getitem__(self, key)
except KeyError:
return self.__missing__(key)
def __missing__(self, key):
if self.default_factory is None:
raise KeyError(key)
self[key] = value = self.default_factory()
return value
def __reduce__(self):
if self.default_factory is None:
args = tuple()
else:
args = self.default_factory,
return type(self), args, None, None, self.items()
def copy(self):
return self.__copy__()
def __copy__(self):
return type(self)(self.default_factory, self)
def __deepcopy__(self, memo):
import copy
return type(self)(self.default_factory,
copy.deepcopy(self.items()))
def __repr__(self):
return 'OrderedDefaultDict(%s, %s)' % (self.default_factory,
OrderedDict.__repr__(self))
Aquí hay otra posibilidad, inspirada en super() Considered Super de Raymond Hettinger , probado en Python 2.7.X y 3.4.X:
from collections import OrderedDict, defaultdict
class OrderedDefaultDict(OrderedDict, defaultdict):
def __init__(self, default_factory=None, *args, **kwargs):
#in python3 you can omit the args to super
super(OrderedDefaultDict, self).__init__(*args, **kwargs)
self.default_factory = default_factory
Si revisas el MRO de la clase (también conocido como help(OrderedDefaultDict)), verás esto:
class OrderedDefaultDict(collections.OrderedDict, collections.defaultdict)
| Method resolution order:
| OrderedDefaultDict
| collections.OrderedDict
| collections.defaultdict
| __builtin__.dict
| __builtin__.object
lo que significa que cuando OrderedDefaultDictse inicializa una instancia de, se difiere al OrderedDictinicio de ', pero éste a su vez llamará a los defaultdictmétodos de ' antes de llamar a __builtin__.dict, que es precisamente lo que queremos.
Si desea una solución simple que no requiera una clase, puede usar o . Si solo obtiene/configura desde algunos lugares, digamos en un bucle, puede configurar fácilmente el valor predeterminado.OrderedDict.setdefault(key, default=None)OrderedDict.get(key, default=None)
totals = collections.OrderedDict()
for i, x in some_generator():
totals[i] = totals.get(i, 0) + x
Es aún más fácil para listas con setdefault:
agglomerate = collections.OrderedDict()
for i, x in some_generator():
agglomerate.setdefault(i, []).append(x)
Pero si lo usas más de unas pocas veces, probablemente sea mejor configurar una clase, como en las otras respuestas.
