将多级列表列表展平为单级

Question

我需要展平多级列表的所有级别：

import itertools

pool = [[[[[0,2],[1,3]],[[3,2],[1,0]]],"PdPd","PrisonerDilemma"], 
[[[[0,3],[1,2]],[[2,3],[1,0]]],"ShSh","StagHunt"],
[[[[1,2],[3,0]],[[3,2],[1,0]]],"ChCh","Chicken"],
[[[[2,1],[0,3]],[[3,1],[0,2]]],"BaBa","Battle"]]

def flat3 (pool):
    for game in pool:
        l = list(itertools.chain.from_iterable(game[0]))
        print(l)

结果：

flat3 (pool)
[[0, 2], [1, 3], [3, 2], [1, 0]]
[[0, 3], [1, 2], [2, 3], [1, 0]]
[[1, 2], [3, 0], [3, 2], [1, 0]]
[[2, 1], [0, 3], [3, 1], [0, 2]]

因此，目标是隔离并返回每个项目中的第一级，仅包含数字，例如：

[[[[[0,2],[1,3]],[[3,2],[1,0]]],"PdPd","PrisonerDilemma"]

然后我需要将所有内容都展平到同一级别，例如：

[0,2,1,3,3,2,1,0]

我知道有很多关于这个主题的材料，我已经找到并尝试了很多方法来做到这一点，但是似乎没有一个可以在一个以上的层次上工作，我想知道是否有一种有效的方法来做到这一点，而不重复多次执行相同的命令。有人可以帮助我吗？

Answer 1

像往常一样，第一步是查看您的数据结构。

el = [
         [
             [[0,2],[1,3]],
             [[3,2],[1,0]]
         ],
         "PdPd",
         "PrisonerDilemma"
     ]

这是您结构中的每个单独元素。我们暂时忽略这个最外层结构几乎肯定不应该是一个列表（对我来说它看起来更像是一个元组），而只关注我们得到的东西。

el[0] = [ [[0, 2], [1, 3]],
          [[3, 2], [1, 0]] ]
el[1] = # a string
el[2] = # another string, we don't care about these

现在我们已将其简化为数字列表的列表。这个我们可以操作。

def flatten(lst):
    for el in lst:
        if isinstance(el, list):  # N.B. this only works for lists, not general
                                  # collections e.g. sets, tuples, dicts, etc...
            # recurse
            yield from flatten(el)
        else:
            # generate
            yield el

您的结果是将此函数应用于最外层列表中每个项目的第一个元素。

result = [flatten(sublst[0]) for sublst in big_list]

请注意，这会创建一个生成器对象列表，需要使用这些对象才能实际生成值。如果出于某种原因您确实需要列表本身，请显式转换为列表：

result = [list(flatten(sublst[0])) for sublst in big_list]