使用itertools处理python嵌套列表问题
-
使用itertools处理python嵌套列表问题
-
当我们想要把一串字符分割成列表的时候
import itertools
a = "1111"
>>>print(list(itertools.chain(a)))
output: ['1', '1', '1', '1']
- 嵌套列表组合成一个列表
import itertools
>>> a = [1, [2], [3], 4, 5]
>>> print(itertools.chain.from_iterable(a))
output: [1, 2, 3, 4, 5]