Python - trier une liste de listes imbriquées
-
07-07-2019 - |
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?
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)
Où 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