2008-11-11 5 views
13

У меня есть вход, состоящий из списка вложенных списков вроде этого:Python - сортировать список вложенных списков

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

Я хочу, чтобы отсортировать этот список на основе суммы всех чисел в вложенных списках. .. поэтому, значения я хочу сортировать по л будет выглядеть следующим образом:

[39, 6, 13, 50] 

Тогда я хочу, чтобы отсортировать на их основе. Таким образом, выход должен быть:

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

Что хорошего в питоническом способе сделать это?

ответ

16

небольшое упрощение и обобщение ответов, представленных до сих пор, используя недавнее дополнение к синтаксис 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) 

Где sum_nested() является:

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

Немного рекурсивная функция будет делать это:

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

Я бы сказал, что `hasattr` является более общим решением в этом случае, чем` isinstance`. – jfs 2008-11-11 20:58:34