繁体   English   中英

有没有办法在嵌套字典理解中使用临时列表变量?

[英]Is there a way to use temporary list variable inside nested dict comprehension?

这是我的字典的样子:

   ranked_imgs = {'d57b26aa-c8f8-4e08-9e8f-8aecd6e952a5': {'small': [],
      'medium': [['https://www.pablo-ruiz-picasso.net/images/works/261.jpg', 1]],
      'large': []},
     'b10ecfdb-f590-4930-9975-aa12dc267f2f': {'small': [],
      'medium': [],
      'large': [['https://www.pablo-ruiz-picasso.net/images/works/3936.jpg', 1]]},......}

这就是我尝试使用字典理解来做的事情,但它失败了,因为临时列表由于 for 循环而被覆盖:

dct = {k:[x for x in vv] for k,v in ranked_imgs.items() for vk,vv in v.items()}

或者像这样的东西:

dct = {k:sum(vv,[]) for k,v in ranked_imgs.items() for vk,vv in v.items()}

这就是代码在理解后的样子:

# ranked_imgs_={}

# for k,v in ranked_imgs.items():
#     lst = []
#     for vk,vv in v.items():
#         for url in vv:
#             lst.append(url)
#     ranked_imgs_[k] = lst

我很想知道是否有一种 pythonic 方法可以使用字典理解来做到这一点!

您预期的 output 是列表的字典,因此您可以使用列表理解作为字典理解的 output 值。 另请注意,您可以迭代v.values()而不是v.items()因为您没有使用键(感谢@MechanicPig 指出):

dct = {k: [url for vv in v.values() for url in vv] for k, v in ranked_imgs.items()}

使用chain.from_iterable可能比@blhsing 的答案好一点,它通过连接v.values()进行迭代:

from itertools import chain
result = {k: list(chain.from_iterable(v.values())) for k, v in ranked_imgs.items()}

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM