如何解决解包 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))
然而,
- 如何对 list of list 列表做同样的事情?
- 如何循环输入以仅分解列表列表?
解决方法
仅当行的第一项是列表时才使用 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']]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。