Python - 排序嵌套列表的列表

12 投票
3 回答
8244 浏览
提问于 2025-04-11 19:48

我有一个输入,它是一个包含嵌套列表的列表,像这样:

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

有没有什么好的方法可以用Python来实现这个呢?

3 个回答

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

撰写回答