Python - 排序嵌套列表的列表
我有一个输入,它是一个包含嵌套列表的列表,像这样:
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]]