python - 嵌套列表理解

Question

我有 2 个列表，A 和 B。

• A 是包含值列表的字典列表。
• B 是一个简单的列表。

需求是将列表B的元素添加到A的字典值中。

下面是代码，

a = [{'a':[1,5]}, {'b' : [6,10]}, {'c' : [11,15]}]
b = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15]

for i in a:
  for k,j in i.items():
      for m,n in enumerate(b):
          if j[0]<= n<=j[1]:
              j.append(n)

print(a)

[{'a':[1,5,1,2,3,4,5]}, {'b' : [6,10,6,7,8,9,10]}, {'c' : 
[11,15,11,12,13,14,15]}]

# tried list comprehension
a= [{k:n} for i in a for k,j in i.items() for m,n in enumerate(b)
if j[0]<= n<=j[1]]

print(a)

[{'a': 1}, {'a': 2}, {'a': 3}, {'a': 4}, {'a': 5}, {'b': 6}, {'b': 7}, 
{'b': 8}, {'b': 9}, {'b': 10}, {'c': 11}, {'c': 12}, {'c': 13}, {'c': 
14}, {'c': 15}]

问题是这是否可以使用列表理解来完成？我已尝试但无法生成所需的输出。

Answer 1

a = [{k: [ x for x in b if v[0] <= x <= v[1] ]} for d in a for k,v in d.items()]