Batch Generator函数在转换为列表时只返回最后一个元素

Question

我有以下代码从项目列表中生成批次：

def batch_generator(items, batch_size):
count = 1
chunk = []

for item in items:
if count % batch_size:
chunk.append(item)
else:
chunk.append(item)
yield chunk
chunk.clear()
count += 1

if len(chunk):
yield chunk

逐一迭代产生预期结果：

for x in batch_generator(range(17), 5):
print(x)
# [0, 1, 2, 3, 4]
# [5, 6, 7, 8, 9]
# [10, 11, 12, 13, 14]
# [15, 16]

然而，当我直接将生成器转换为列表时，只会多次返回最后一个元素！

list(batch_generator(range(17), 5))
# [[15, 16], [15, 16], [15, 16], [15, 16]]

而一个简单的生成器转换为列表就可以了：

list(([i,i*2,i*3] for i in range(5)))
# [[0, 0, 0], [1, 2, 3], [2, 4, 6], [3, 6, 9], [4, 8, 12]]

为什么会发生这种情况？

Answer 1

chunk.clear()就是这里的问题。一天结束时，返回的列表是多次返回的同一列表。

用CCD_ 3代替CCD_。这样chunk将是列表的不同实例：

Answer 2

您可以使用chunk[:]或list(chunk)生成区块的副本。而不仅仅是yield chunk。