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

Это будет работать для вложенного словаря с любыми типами данных.