将嵌套列表更改为多个单个列表

And*_*rew 3 python list nested-lists

如果我有一个清单:

[[ A, B, C ], 10, [ 1, 2, 3 ], F]
Run Code Online (Sandbox Code Playgroud)

将其更改为的最佳方法是什么:

[ A, 10, 1, F]
[ B, 10, 2, F]
[ C, 10, 3, F]
Run Code Online (Sandbox Code Playgroud)

嵌套列表的长度是一致的:len(list[0]) == len(list[2]),但len(list[0])可能是3, 2, 4,等等。

cs9*_*s95 6

让我们itertools在这里试试:

from itertools import cycle

list(zip(*(iter(x) if isinstance(x, list) else cycle([x]) for x in l)))
# [('A', 10, 1, 'F'), ('B', 10, 2, 'F'), ('C', 10, 3, 'F')]
Run Code Online (Sandbox Code Playgroud)

请注意,如果子列表大小不等,这不会出错 - 输出的大小等于最短子列表的大小。


这个怎么运作

遍历列表,如果项目是列表,则将其转换为带有 的迭代器iter,否则,如果标量将其转换为循环迭代器(无限重复单个值)并将它们压缩在一起并列出。

步骤 0

 A     10    1      F
 B           2   
 C           3
l[0]  l[1]  l[2]  l[3]
Run Code Online (Sandbox Code Playgroud)

第1步

(iter(x) if isinstance(x, list) else cycle([x]) for x in l)

 A     10    1      F
 B     10    2      F
 C     10    3      F
l[0]  l[1]  l[2]  l[3]
Run Code Online (Sandbox Code Playgroud)

第2步

list(zip(*_))
[('A', 10, 1, 'F'), ('B', 10, 2, 'F'), ('C', 10, 3, 'F')]
Run Code Online (Sandbox Code Playgroud)