我想在字典中打印字典中的数字:
frame = ['frame1', 'frame2', 'frame3', 'frame4', 'frame5', 'frame6']
dx = [12, 34, 55, 66, 78, 65, 56, 234, 876, 999, 345, 945]
pd = ['pd1', 'pd2']
data = {}
dt = {}
k=0
for i in pd :
dt[i] = []
for j in frame :
data[j] = dt
data[j][i] = dx[k]
k+=1
对于这段代码,它只保留最后的值:
{'frame1': {'pd1': 65, 'pd2': 945}, 'frame2': {'pd1': 65, 'pd2': 945}, 'frame3': {'pd1': 65, 'pd2': 945}, 'frame4': {'pd1': 65, 'pd2': 945}, 'frame5': {'pd1': 65, 'pd2': 945}, 'frame6': {'pd1': 65, 'pd2': 945}}
这是我想要的结果:
{'frame1': {'pd1': 12, 'pd2': 56}, 'frame2': {'pd1': 34, 'pd2': 234}, 'frame3': {'pd1': 55, 'pd2': 876}, 'frame4': {'pd1': 66, 'pd2': 999}, 'frame5': {'pd1': 78, 'pd2': 345}, 'frame6': {'pd1': 65, 'pd2': 945}}
我怎样才能做到这一点?
看起来所有的字典都有共享内存
可以使用字典推导式构造字典:
{f: dict(zip(pd, vals)) for f, vals in zip(frame, zip(dx[:len(dx)//2], dx[len(dx)//2:]))}
这个输出:
{
'frame1': {'pd1': 12, 'pd2': 56}, 'frame2': {'pd1': 34, 'pd2': 234},
'frame3': {'pd1': 55, 'pd2': 876}, 'frame4': {'pd1': 66, 'pd2': 999},
'frame5': {'pd1': 78, 'pd2': 345}, 'frame6': {'pd1': 65, 'pd2': 945}
}
您可以从嵌套的字典理解中获得它,但是它有点长:
{f: d for f, d in zip(frame, {pd[0]: i, pd[1]: j} for i, j in zip(dx[:len(frame)], dx[len(frame):]))}
一个小的简化:
dict(zip(frame, (dict(zip(pd, x)) for x in zip(dx[:len(frame)], dx[len(frame):]))))
输出:
{'frame1': {'pd1': 12, 'pd2': 56},
'frame2': {'pd1': 34, 'pd2': 234},
'frame3': {'pd1': 55, 'pd2': 876},
'frame4': {'pd1': 66, 'pd2': 999},
'frame5': {'pd1': 78, 'pd2': 345},
'frame6': {'pd1': 65, 'pd2': 945}}
应该可以了
data = {}
k=0
for i in pd:
for j in frame :
# use setdefault to initialize inner dictionary
data.setdefault(j, {})[i] = dx[k]
k+=1
data
{'frame1': {'pd1': 12, 'pd2': 56},
'frame2': {'pd1': 34, 'pd2': 234},
'frame3': {'pd1': 55, 'pd2': 876},
'frame4': {'pd1': 66, 'pd2': 999},
'frame5': {'pd1': 78, 'pd2': 345},
'frame6': {'pd1': 65, 'pd2': 945}}
您的问题是您在外部for循环的每次迭代中重置dt
。要得到想要的输出,只需执行以下命令
for i in pd:
for j in frame:
if j not in data:
data[j] = dict()
data[j][i] = dx[k]
k += 1
你可以把它压缩成一个像这样的列表推导式:
data = {f: {p: dx[j * len(frame) + i] for j, p in enumerate(pd)} for i, f in enumerate(frame)}
此压缩遍历frames
,并且对于frames
中的每个项,遍历pd
。对于pd
中的每个项,我们在向大字典中添加项时向字典中添加项。dx
添加到字典中的值是基于pd
的第j项和frames
的第i项。
更好的变量名有助于跟踪正在发生的事情。可读性与单行列表推导比较。原来的代码在每一帧中插入了相同的修改过的字典。
from pprint import pprint
frames = ['frame1', 'frame2', 'frame3', 'frame4', 'frame5', 'frame6']
dx = [12, 34, 55, 66, 78, 65, 56, 234, 876, 999, 345, 945]
pds = ['pd1', 'pd2']
# Initialize each frame with a *different* empty dictionary
data = {frame:{} for frame in frames}
# Distribute dx across the frames and pds:
idx = iter(dx)
for pd in pds:
for frame in frames:
data[frame][pd] = next(idx)
pprint(data)
输出:
{'frame1': {'pd1': 12, 'pd2': 56},
'frame2': {'pd1': 34, 'pd2': 234},
'frame3': {'pd1': 55, 'pd2': 876},
'frame4': {'pd1': 66, 'pd2': 999},
'frame5': {'pd1': 78, 'pd2': 345},
'frame6': {'pd1': 65, 'pd2': 945}}