python - 从字典键的子集创建命名元组

Question

这个问题已经有答案了:

Creating a namedtuple object using only a subset of arguments passed (3 个回答)

已关闭 5 年前。

通过以下测试的最Pythonic方法是什么:

def test_named_tuple_from_map():
    from collections import namedtuple
    NT = namedtuple('nt', ['a', 'b', 'c'])
    source = {'a': 1, 'b': 2, 'd': 4, 'c': 3}
    new = NT(**source)
    assert new == NT(a=1,b=2,c=3)

我想要这个的原因是这样我可以挑选出 dict 数据的子集并作为不可变数据传递。

Answer 1

您可以简单地使用 namedtuple 的属性 _fields 并相应地过滤字典:

def test_named_tuple_from_map():
    from collections import namedtuple
    NT = namedtuple('nt', ['a', 'b', 'c'])
    source = {'a': 1, 'b': 2, 'd': 4, 'c': 3}

    new = NT(**{k: v for k, v in source.items() if k in NT._fields})

    assert new == NT(a=1, b=2, c=3)