Python - 如何将多维列表排序为二维列表?

Mar*_*off 5 python list

我如何将多维列表排序到二维列表?

多维输入: [8, [6, 7, [-1], [4, [[10]]], 2], 1]

期望的二维输出: [[8, 1], [6, 7, 2], [-1, 4], [], [10]]

所有相同的深度列表项都需要在同一个列表中.

Dan*_*ejo 2

这个想法基本上与 @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)
Run Code Online (Sandbox Code Playgroud)

输出

[[8, 1], [6, 7, 2], [-1, 4], [], [10]]
Run Code Online (Sandbox Code Playgroud)