Python - сортировка списка вложенных списков
-
07-07-2019 - |
Вопрос
У меня есть вход, состоящий из списка вложенных списков, подобных этому:
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:
>>> 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]]
Другие советы
Маленькая рекурсивная функция сделает это:
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
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
Не связан с StackOverflow