温馨提示×

Python chunk函数能返回迭代器吗

小樊
82
2024-07-18 15:57:29
栏目: 编程语言

是的,Python中的chunk函数可以返回一个迭代器对象。通过使用chunk函数,可以将一个可迭代对象分割成指定大小的块,然后返回一个迭代器,每次迭代返回一个块。这样可以方便对大型数据集进行分块处理。以下是一个示例代码:

from itertools import islice

def chunk(iterable, size):
    it = iter(iterable)
    return iter(lambda: tuple(islice(it, size)), ())

data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

for chunked_data in chunk(data, 3):
    print(chunked_data)

输出结果为:

(1, 2, 3)
(4, 5, 6)
(7, 8, 9)
(10,)

0