2008-11-11 35 views
13

tengo entrada que consiste en una lista de listas anidadas así:Python - ordenar una lista de listas anidadas

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

Quiero ordenar esta lista en base a la suma de todos los números en las listas anidadas. .. así, los valores que desea ordenar de l se vería así:

[39, 6, 13, 50] 

Entonces desea ordenar en base a estos. Entonces la salida debería ser:

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

¿Qué es una buena manera pitonica de hacer esto?

Respuesta

16

Una ligera simplificación y generalización de las respuestas dadas hasta ahora, utilizando una adición reciente a 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]] 
5
l.sort(key=sum_nested) 

Dónde 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 
12

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 
+1

Diría que 'hasattr' es una solución más general en este caso que' isinstance'. – jfs

Cuestiones relacionadas