Question

J'ai une entrée composée d'une liste de listes imbriquées comme celle-ci:

l = [[[[[39]]]], [1, 2, 3], [4, [5, 3], 1], [[[[8, 9], 10], 11], 12]]

Je veux trier cette liste en fonction de la somme de tous les nombres des listes imbriquées ... les valeurs que je veux trier par l ressemblent donc à ceci:

[39, 6, 13, 50]

Ensuite, je veux trier en fonction de ceux-ci. Donc, le résultat devrait être:

[[1, 2, 3], [4, [5, 3], 1], [[[[39]]]], [[[[8, 9], 10], 11], 12]]

Qu'est-ce qu'une bonne façon pythonique de faire cela?

Était-ce utile?

La solution

Une légère simplification et généralisation des réponses fournies jusqu'à présent, en utilisant un ajout récent à la syntaxe de python:

>>> l = [[[[[39]]]], [1, 2, 3], [4, [5, 3], 1], [[[[8, 9], 10], 11], 12]]
>>> def asum(t): return sum(map(asum, t)) if hasattr(t, '__iter__') else t
...
>>> sorted(l, key=asum)
[[1, 2, 3], [4, [5, 3], 1], [[[[39]]]], [[[[8, 9], 10], 11], 12]]

Autres conseils

Une petite fonction récursive le ferait:

def asum(a):
    if isinstance(a, list):
        return sum(asum(x) for x in a)
    else:
        return a

l = [[[[[39]]]], [1, 2, 3], [4, [5, 3], 1], [[[[8, 9], 10], 11], 12]]
l.sort(key=asum)
print l
l.sort(key=sum_nested)

sum_nested () est:

def sum_nested(astruct):
    try: return sum(map(sum_nested, astruct))
    except TypeError:
        return astruct


assert sum_nested([[([8, 9], 10), 11], 12]) == 50
Licencié sous: CC-BY-SA avec attribution
Non affilié à StackOverflow
scroll top