How to create a list of values in a dictionary comprehension in Python

后端 未结 2 1936
太阳男子
太阳男子 2021-01-24 07:23

Taking a very simple example of looping over a sentence and creating a dictionary which maps {x:y}, where x is a key representing the length of the wor

2条回答
  •  一向
    一向 (楼主)
    2021-01-24 07:54

    Sure, you can, using sorted + groupby, but it doesn't look great.

    from itertools import groupby
    d = dict([(k, list(g)) for k, g in groupby(sorted(mywords.split(), key=len), key=len)])
    
    print(d)
    {2: ['be', 'be'],
     3: ['May', 'and'],
     4: ['your', 'your'],
     5: ['short'],
     6: ['coffee', 'strong', 'Monday']}
    

    P.S., Here's my answer (using defaultdict that I recommend over this) to the original question.

提交回复
热议问题