如何复制一个字典并在一行代码中修改它

Question

我经常需要创建一个或两个项目彼此不同的词汇.这是我通常做的事情:

setup1 = {'param1': val1, 
            'param2': val2,
            'param3': val3,
            'param4': val4,
            'paramN': valN}

setup2 = copy.deepcopy(dict(setup1))
setup2.update({'param1': val10, 
                   'param2': val20})

程序中有一个点setup2是相同副本的事实setup1让我感到紧张,因为我担心在程序生命的某个时刻,两条线可能会分开,这对于太多的bug来说是一个滑坡.

理想情况下,我希望能够在一行代码中完成此操作(类似这样):

setup2 = dict(setup1).merge({'param1': val10, 
                        'param2': val20})

当然,我可以使用分号将两个命令压缩到一个物理行中,但这对我来说看起来很难看.还有其他选择吗？

Answer 1

在我看来,最简单的方法是这样的:

new_dict = {**old_dict, 'changed_val': value, **other_new_vals_as_dict}

Answer 2

setup2 = dict(setup1.items() + {'param1': val10, 'param2': val20}.items())

这样,如果添加了新密钥,则会setup1替换旧密钥/值对.

Answer 3

您可以在字典构造函数中使用关键字参数进行更新

new = dict(old, a=1, b=2, c=3)

# You can also unpack your modifications
new = dict(old, **mods)

这相当于:

new = old.copy()
new.update({"a": 1, "b": 2, "c": 3})

资源

注意:dict.copy()创建浅表副本.

Answer 4

解

为此构建一个功能.

当您在代码中使用它时,您的意图会更清晰,您可以在一个地方处理复杂的决策(例如,深拷贝和浅拷贝).

def copy_dict(source_dict, diffs):
    """Returns a copy of source_dict, updated with the new key-value
       pairs in diffs."""
    result=dict(source_dict) # Shallow copy, see addendum below
    result.update(diffs)
    return result

现在副本是原子的,假设没有线程涉及:

setup2=copy_dict(setup1, {'param1': val10, 'param2': val20})

附录 - 深层复制

对于基元(整数和字符串),不需要深层复制:

>>> d1={1:'s', 2:'g', 3:'c'}
>>> d2=dict(d1)
>>> d1[1]='a'
>>> d1
{1: 'a', 2: 'g', 3: 'c'}
>>> d2
{1: 's', 2: 'g', 3: 'c'}

如果您需要深层副本,请使用该copy模块:

result=copy.deepcopy(source_dict) # Deep copy

代替:

result=dict(setup1)               # Shallow copy

确保字典中的所有对象都支持深层复制(任何可以pickled做的对象).

Answer 5

setup2 = dict((k, {'param1': val10, 'param2': val20}.get(k, v))
              for k, v in setup1.iteritems())

这仅在更新字典的所有键都已包含在内时才有效setup1.

如果你的所有键都是字符串,你也可以这样做

setup2 = dict(setup1, param1=val10, param2=val20)

Answer 6

从Python 3.9开始，您可以使用管道命令（例如first_dic | second_dic）来合并字典；它还可用于通过首先传递原始字典并将更新作为第二个字典来返回新的更新字典：

setup2 = setup1 | {'param1': val10, 'param2': val20}