Question

我的输入包含一个嵌套列表列表，如下所示：

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方式？

Answer 1

使用最近添加的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]]

Answer 2

一个小的递归函数会这样做：

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

Answer 3

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

Python - 对嵌套列表进行排序

3 个答案: