2008-11-11 7 views
13

Ich habe Eingabe einer Liste von verschachtelten Listen wie folgt aus:Python - sortieren eine Liste von verschachtelten Listen

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

ich diese Liste sortiert werden soll, basierend auf der Summe aller Zahlen in den verschachtelten Listen. Die Werte, nach denen ich sortieren möchte, würden folgendermaßen aussehen:

Dann möchte ich basierend auf diesen sortieren. So sollte die Ausgabe sein:

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

Was ist eine schöne pythonische Art, dies zu tun?

Antwort

16

eine leichte Vereinfachung und Verallgemeinerung zu den bisher gegebenen Antworten, unter Verwendung eines kürzlich zusätzlich zu Python-Syntax:

>>> 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) 

Wo sum_nested() ist:

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

Eine wenig rekursive Funktion würde es tun:

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

Ich würde sagen, dass "hasattr" in diesem Fall eine allgemeinere Lösung ist als "isinstance". – jfs