1

我有两个具有相同键的字典。每个项目都是一个ndarray。

from numpy import zeros, random
from collections import namedtuple

PhaseAmplitude = namedtuple('PhaseAmplitude','phase amplitude')
dict_keys = {'K1','K2', 'K3'}

J1 = dict.fromkeys(dict_keys, zeros((2,2,2,2)))
U1 = dict.fromkeys(dict_keys, PhaseAmplitude(phase = zeros((2,2)),
                                             amplitude = zeros((2,2))))

for iFld in dict_keys:
    U1[iFld] = U1[iFld]._replace(phase = random.random_sample((2,2)), 
                                 amplitude = random.random_sample((2,2)))

我想使用第二个字典中的相应项目修改第一个字典中的每个项目：

for iFld in dict_keys:  
    J1[iFld][0,0,:,:] += U1[iFld].phase
    J1[iFld][0,1,:,:] += U1[iFld].amplitude

我希望得到它J1[iFld][0,0,:,:] = U1[iFld].phase，J1[iFld][0,1,:,:] = U1[iFld].amplitude但我得到J1[iFld]的是对所有人都是一样的，iFld并且等于所有iFld键的总和U1（当然要跟踪phase和amplitude字段U1）。

对我来说，这看起来像是一个错误，但我只使用 Python 一个月左右（从 matlab 切换）所以我不确定。

问题：这是预期的行为还是错误？为了获得我想要的行为，我应该在我的代码中进行哪些更改？

注意：我选择了的维数dict_keys和以反映我的特殊情况。J1U1

4

1 回答 1

1

这不是一个错误，尽管它是一个非常常见的陷阱，会在几种不同的情况下出现。 dict.fromkeys创建一个新字典，其中所有值都是同一个对象。这对于不可变类型（例如int, str）非常有效，但对于可变类型，您可能会遇到问题。

例如：

>>> import numpy as np
>>> d = dict.fromkeys('ab', np.zeros(2))
>>> d
{'a': array([ 0.,  0.]), 'b': array([ 0.,  0.])}
>>> d['a'][1] = 1
>>> d
{'a': array([ 0.,  1.]), 'b': array([ 0.,  1.])}

这是因为：

>>> d['a'] is d['b']
True

在这种情况下，使用 dict 理解来构建字典：

J1 = {k: zeros((2,2,2,2)) for k in dict_keys}

（或者，pre-python2.7）：

J1 = dict((k, zeros((2,2,2,2))) for k in dict_keys)
于 2016-03-09T17:11:37.783 回答