Créer ou ajouter à une liste dans un dictionnaire - cela peut-il être raccourcie?

Peut ce code Python être raccourcie et toujours être lisible à l'aide de itertools et de jeux?

result = {}
for widget_type, app in widgets:
    if widget_type not in result:
        result[widget_type] = []
    result[widget_type].append(app)

Je peux penser de cela seulement:

widget_types = zip(*widgets)[0]
dict([k, [v for w, v in widgets if w == k]) for k in set(widget_types)])
InformationsquelleAutor culebrón | 2010-11-10
  1. 47

    Vous pouvez utiliser un defaultdict(list).

    from collections import defaultdict

result = defaultdict(list)
for widget_type, app in widgets:
    result[widget_type].append(app)
    • N'a jamais su de cette. Cool!
    InformationsquelleAutor Mark Byers
  2. 86

    Une alternative à defaultdict est d'utiliser le setdefault méthode de dictionnaires standard:

     result = {}
 for widget_type, app in widgets:
     result.setdefault(widget_type, []).append(app)

    Cela repose sur le fait que les listes sont mutables, donc ce qui est retourné à partir de setdefault est la même liste que celle dans le dictionnaire, donc vous pouvez y ajouter.

    • Et c'est mon Python pépite à apprendre pour la journée. Merci, Daniel. 🙂
    InformationsquelleAutor Daniel Roseman
  3. 4

    peut être un peu lent mais fonctionne

    result = {}
for widget_type, app in widgets:
    result[widget_type] = result.get(widget_type, []) + [app]
    • pourquoi en serait-il plus lent que les autres solutions?
    • Il serait plus lente, car la méthode concat opérateur (+) crée une nouvelle liste à chaque fois, alors que append() ajoute l'élément à la liste existante.
    InformationsquelleAutor user1139002