Питонический способ преобразования словаря в namedtuple или другой Хэшируемый dict-подобный?
у меня есть словарь, как:
d = {'a': 1, 'b': 2, 'c': 3, 'd': 4}
которые я хотел бы преобразовать в namedtuple. Мой текущий подход заключается в следующем коде
namedTupleConstructor = namedtuple('myNamedTuple', ' '.join(sorted(d.keys())))
nt= namedTupleConstructor(**d)
которая производит
myNamedTuple (a=1, b=2, c=3, d=4)
Это отлично работает для меня (я думаю), но мне не хватает встроенного, такого как...
nt = namedtuple.from_dict() ?
UPDATE: как обсуждалось в комментариях, моя причина желания преобразовать мой словарь в namedtuple так что она становится hashable, но еще в целом полезная, как дикт.
4 ответов
чтобы создать подкласс, вы можете просто передать ключи dict напрямую:
MyTuple = namedtuple('MyTuple', sorted(d))
теперь, чтобы создать экземпляры из этого Дикта или других диктовок с соответствующими ключами:
my_tuple = MyTuple(**d)
внимание: namedtuples сравнить на значения (заказал). Они предназначены для замены обычных кортежей с доступом к именованному атрибуту в качестве добавленной функции. имена полей не будут учитываться при равенстве сравнения. Это отличается от dict
сравнения равенства, которые учитывают ключи, и это может быть не то, что вы хотели и не ожидали от namedtuple
тип!
если у вас есть только один дикт, а не куча диктов, разделяющих один и тот же набор ключей, то нет смысла создавать этот namedtuple в первую очередь. Вместо этого следует использовать объект пространства имен:
>>> from types import SimpleNamespace
>>> SimpleNamespace(**d)
namespace(a=1, b=2, c=3, d=4)
для хэшируемого" attrdict", как рецепт, проверьте замороженный box:
>>> from box import Box
>>> b = Box(d, frozen_box=True)
>>> hash(b)
7686694140185755210
>>> b.a
1
>>> b['a']
1
вы можете использовать эту функцию для обработки вложенных словарей:
def create_namedtuple_from_dict(obj):
if isinstance(obj, dict):
fields = sorted(obj.keys())
namedtuple_type = namedtuple(
typename='GenericObject',
field_names=fields,
rename=True,
)
field_value_pairs = OrderedDict(
(str(field), create_namedtuple_from_dict(obj[field]))
for field in fields
)
try:
return namedtuple_type(**field_value_pairs)
except TypeError:
# Cannot create namedtuple instance so fallback to dict (invalid attribute names)
return dict(**field_value_pairs)
elif isinstance(obj, (list, set, tuple, frozenset)):
return [create_namedtuple_from_dict(item) for item in obj]
else:
return obj
зацени вот это:
def fill_tuple(NamedTupleType, container):
if container is None:
args = [None] * len(NamedTupleType._fields)
return NamedTupleType(*args)
if isinstance(container, (list, tuple)):
return NamedTupleType(*container)
elif isinstance(container, dict):
return NamedTupleType(**container)
else:
raise TypeError("Cannot create '{}' tuple out of {} ({}).".format(NamedTupleType.__name__, type(container).__name__, container))
исключения для неправильных имен или недопустимого количества аргументов обрабатываются __init__
of namedtuple
.
тест с py.тест:
def test_fill_tuple():
A = namedtuple("A", "aa, bb, cc")
assert fill_tuple(A, None) == A(aa=None, bb=None, cc=None)
assert fill_tuple(A, [None, None, None]) == A(aa=None, bb=None, cc=None)
assert fill_tuple(A, [1, 2, 3]) == A(aa=1, bb=2, cc=3)
assert fill_tuple(A, dict(aa=1, bb=2, cc=3)) == A(aa=1, bb=2, cc=3)
with pytest.raises(TypeError) as e:
fill_tuple(A, 2)
assert e.value.message == "Cannot create 'A' tuple out of int (2)."
хотя мне нравится ответ @fuggy_yama, прежде чем читать его, я получил свою собственную функцию, поэтому я оставляю его здесь, чтобы показать другой подход. Он также обрабатывает вложенные namedtuples
def dict2namedtuple(thedict, name):
thenametuple = namedtuple(name, [])
for key, val in thedict.items():
if not isinstance(key, str):
msg = 'dict keys must be strings not {}'
raise ValueError(msg.format(key.__class__))
if not isinstance(val, dict):
setattr(thenametuple, key, val)
else:
newname = dict2namedtuple(val, key)
setattr(thenametuple, key, newname)
return thenametuple