13

我有一个由嵌套列表组成的输入,如下所示:

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

这样做的好pythonic方式是什么?

4

3 回答 3

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]]
于 2008-11-11T13:26:37.940 回答
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
于 2008-11-11T07:19:20.130 回答
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
于 2008-11-11T07:18:02.817 回答