Question

我有一个Python字典，如下所示：

d = {'1': {'1': 3, '2': 1, '3': 1, '4': 4, '5': 2, '6': 3},
     '2': {'1': 3, '2': 3, '3': 1, '4': 2},
     '3': {'1': 1, '2': 1, '3': 3, '4': 2, '5': 1, '6': 1, '7': 1},
     '4': {'1': 1, '2': 1, '3': 3, '4': 2, '5': 1, '6': 1, '7': 1}}

我对字典有此操作：

D = {}
for ko, vo in d.items():
  for ki, vi in vo.items():
    for i in range(vi):
      D[f'{ko}_{ki}_{i}'] = someFunc(ko, ki, i)

我想将其翻译成具有字典理解力的一个衬里，如下所示：

D = {f'{ko}_{ki}_{i}': someFunc(ko, ki, i) for i in range(vi) for ki, vi in vo.items() for ko, vo in d.items()}

但是我得到一个错误

NameError：未定义名称'vi'

有人可以帮助我以正确的语法实现这一目标吗？

Answer 1

循环的顺序必须颠倒。

这是您要寻找的：

D = {f'{ko}_{ki}_{i}': someFunc(ko, ki, i) for ko, vo in d.items() for ki, vi in vo.items() for i in range(vi)  }

Answer 2

列表理解中的for子句应以与相同的for循环代码相同的顺序出现。唯一可以“移动”的是，最里面的赋值被开头的表达式替换。

有关详细信息，请参见https://treyhunner.com/2015/12/python-list-comprehensions-now-in-color/。

Python中的3级嵌套字典理解

2 个答案: