解包 list of list

问题描述

我在程序中使用的当前输入是一个坐标列表：

data = [["D_02","10","20"],["X_03","20"]]

为了便于使用和方便，我也希望能够传递一个列表列表：

have：

data = [[["B_01"],["10"],["10","20"]],["D_02","20"]]

然后能够“爆炸”它以使其如下所示：

want：

data = [["B_01","10"],["B_01","20","20"]]

我知道可以使用 product 包的 itertools 函数对列表列表执行此操作：

import itertools
list(itertools.product(*data))

然而，

1. 如何对 list of list 列表做同样的事情？
2. 如何循环输入以仅分解列表列表？

解决方法

仅当行的第一项是列表时才使用 itertools.product：

final = []
for l in data:
    if isinstance(l[0],list):  # <- ["B01"]
        for c in itertools.product(*l[1:]):
            final.append(l[0] + list(c))
    else:
        final.append(l)

>>> final
[['B_01','10','10'],['B_01','20'],'20',['D_02',['X_03','20']]