说明
看文档发现一个有趣的应用(利用zip函数)
例如[1, 2, 3, 4] --> [(1, 2), (3, 4)],拆分成长度为2的数据块
Code
>>> a = [1,2,3,4]
>>> length = 2
>>> chunks_len_2 = zip(*[iter(a)] * length)
>>> result = list(chunks_len_2)
>>> result
[(1, 2), (3, 4)]
原理: zip(*iterables)
https://docs.python.org/3/library/functions.html#zip
def zip(*iterables):
# zip('ABCD', 'xy') --> Ax By
sentinel = object()
iterators = [iter(it) for it in iterables]
while iterators:
result = []
for it in iterators:
elem = next(it, sentinel)
if elem is sentinel:
return
result.append(elem)
yield tuple(result)