python - 如何限制块功能？

Question

我正在尝试创建一个块函数，其想法是它为列表中特定数量的元素重复某些任务（用 表示n）

一个例子是如果n=5then 只打印前 5 个元素。

这是我的尝试：

def chunks(lst, n):
    """Yield successive n-sized chunks from lst."""
    for i in range(0, len(lst), n):
        yield lst[i:i + n]
        

a_random_list=[1,2,3,4,5,6,7,8,9,10]
        
just_some_chunks = list(chunks(a_random_list, 5))

print(just_some_chunks)

控制台输出：

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

预期输出：

[[1,2,3,4,5]]

我如何调整chunks()功能以检索第一个n元素？

标签： pythonbatch-processing