如何将多维列表排序为二维列表?
多维输入:[8, [6, 7, [-1], [4, [[10]]], 2], 1]
期望的二维输出:[[8, 1], [6, 7, 2], [-1, 4], [], [10]]
所有相同的深度列表项都需要在同一个列表中。
最佳答案
这个想法与@TerryA 回答中的基本相同,但使用 setdefault并在 for 循环结束时检查是否添加了深度内容:
lst = [8, [6, 7, [-1], [4, [[10]]], 2], 1]
def depths(l):
def flatten(l, start=0, depth={}):
for e in l:
if isinstance(e, list):
flatten(e, start=start + 1, depth=depth)
else:
depth.setdefault(start, []).append(e)
if start not in depth:
depth[start] = []
d = {}
flatten(l, depth=d)
return [d[i] for i in range(max(d) + 1)]
result = depths(lst)
print(result)
输出
[[8, 1], [6, 7, 2], [-1, 4], [], [10]]
关于Python - 如何将多维列表排序为二维列表?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53632917/