Pythonic способ конвертировать словарь в namedtuple или другой хэшируемый dict-like?
У меня есть словарь вроде:
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, так это то, что он становится хешируемым, но все же обычно полезным, как dict.
Ответы
Ответ 1
Чтобы создать подкласс, вы можете просто передать ключи dict напрямую:
MyTuple = namedtuple('MyTuple', sorted(d))
Теперь, чтобы создать экземпляры из этого или другого, с соответствующими ключами:
my_tuple = MyTuple(**d)
Осторожно: именованные кортежи сравниваются только по значениям (упорядочены). Они предназначены для замены обычных кортежей с доступом к именованным атрибутам в качестве дополнительной функции. Имена полей не будут учитываться при сравнении на равенство. Это отличается от dict
сравнения равенства, которые делают принимать во внимание ключей, и она не может быть то, что вы хотите и не ожидали от namedtuple
типа!
Если у вас есть только один дикт, а не куча диктов, использующих один и тот же набор ключей, то нет смысла создавать этот именованный кортеж в первую очередь. Вы должны просто использовать объект пространства имен вместо:
>>> from types import SimpleNamespace
>>> SimpleNamespace(**d)
namespace(a=1, b=2, c=3, d=4)
Для хэшированного рецепта типа "attrdict" проверьте замороженную коробку:
>>> from box import Box
>>> b = Box(d, frozen_box=True)
>>> hash(b)
7686694140185755210
>>> b.a
1
>>> b['a']
1
Ответ 2
Вы можете использовать эту функцию для обработки вложенных словарей:
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
Ответ 3
Проверьте это:
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__
из namedtuple
.
Тест с py.test:
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)."
Ответ 4
Хотя мне нравится ответ @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
Ответ 5
from collections import namedtuple
nt = namedtuple('x', d.keys())(*d.values())
Ответ 6
def toNametuple(dict_data):
return namedtuple(
"X", dict_data.keys()
)(*tuple(map(lambda x: x if not isinstance(x, dict) else toNametuple(x), dict_data.values())))
d = {
'id': 1,
'name': {'firstName': 'Ritesh', 'lastName':'Dubey'},
'list_data': [1, 2],
}
obj = toNametuple(d)
Доступ как obj.name.firstName
, obj.id
Это будет работать для вложенного словаря с любыми типами данных.