抄一个 batch 函数

需求:将一个数组/可迭代的东西分成 batch

实现:三行代码

从 StackOverflow 上抄的

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

蛮好用的

留下评论