Python - сортировать список вложенных списков

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

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

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

[39, 6, 13, 50]

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

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

Какой хороший питонический способ сделать это?


python list sorting nested-lists
person Claudiu    schedule 11.11.2008    source источник

Ответы (3)


arrow_upward
16
arrow_downward

Небольшое упрощение и обобщение ответов, предоставленных до сих пор, с использованием недавнего дополнения к синтаксису 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]]
person Alex Coventry    schedule 11.11.2008

arrow_upward
12
arrow_downward

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

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
person Greg Hewgill    schedule 11.11.2008
comment
Я бы сказал, что hasattr в этом случае является более общим решением, чем isinstance. - person jfs; 11.11.2008

arrow_upward
5
arrow_downward 

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
person jfs    schedule 11.11.2008