需求:将一个数组/可迭代的东西分成batch
实现:三行代码
从StackOverflow上抄的
def chunk(it, size:int):
it = iter(it)
return iter(lambda: tuple(islice(it, size)), ())
蛮好用的
需求:将一个数组/可迭代的东西分成batch
实现:三行代码
从StackOverflow上抄的
def chunk(it, size:int):
it = iter(it)
return iter(lambda: tuple(islice(it, size)), ())
蛮好用的
发表回复