Python: ordena una lista de listas anidadas
sorting nested-lists (3)
Tengo entrada que consiste en una lista de listas anidadas como esta:
l = [[[[[39]]]], [1, 2, 3], [4, [5, 3], 1], [[[[8, 9], 10], 11], 12]]
Quiero ordenar esta lista en función de la suma de todos los números en las listas anidadas ... así que los valores que quiero ordenar por l serían los siguientes:
[39, 6, 13, 50]
Entonces quiero ordenar basado en estos. Entonces la salida debería ser:
[[1, 2, 3], [4, [5, 3], 1], [[[[39]]]], [[[[8, 9], 10], 11], 12]]
¿Cuál es una buena manera pitonica de hacer esto?
l.sort(key=sum_nested)
Donde sum_nested()
es:
def sum_nested(astruct):
try: return sum(map(sum_nested, astruct))
except TypeError:
return astruct
assert sum_nested([[([8, 9], 10), 11], 12]) == 50
Una pequeña función recursiva lo haría:
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
Una ligera simplificación y generalización de las respuestas proporcionadas hasta ahora, usando una adición reciente a la sintaxis 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]]