Python 使用 itertools 方法将列表拆分成块

在处理大型列表时，我们经常需要将其划分为更小的块以提高处理效率。Python的itertools库提供了一种简单的方法来完成这个任务。

itertools 介绍

itertools是Python标准库中的一个模块，提供了一些用于生成迭代器的工具。使用itertools，我们可以更容易地处理和操作迭代器。

其中一个有用的工具是itertools.groupby()方法，它可以将迭代器分组并返回一个子列表。在这里，我们可以使用该方法将大型列表分成更小的块。

示例代码

import itertools

def chunkify(lst, n):
    """
    将列表分成 n 个块的函数。
  
    参数：
    lst: 需要分块的列表
    n: 分块的数量
    
    返回：
    一个包含 n 个子列表的列表
    """
    # 使用 grouper() 方法将列表分组，每个块的大小都是 ceil(len(lst)/n)
    lst_length = len(lst)
    chunk_size = -(-lst_length // n) # 等同于 int(ceil(lst_length / n))
    return [list(filter(None, chunk)) for chunk in itertools.zip_longest(*[iter(lst)]*chunk_size)]

# 示例用法
lst = list(range(1, 10)) # [1, 2, 3, 4, 5, 6, 7, 8, 9]
n = 3
chunks = chunkify(lst, n)
print(chunks)

上述代码定义了一个函数chunkify()，它接受一个列表和一个整数作为参数，将列表分成n个块并返回一个包含n个子列表的列表。

我们使用itertools.zip_longest()方法将列表分组，并将每个块的大小设置为ceil(len(lst)/n)，也就是通过向上取整来计算每个块的大小。在函数的末尾，我们使用filter()方法来移除返回的子列表中的空值。

在上述代码中，我们还使用列表推导式来将zip_longest()返回的元组转换为列表。

总结

在Python中，使用itertools库中的函数可以更容易地处理和操作迭代器。itertools.groupby()方法是一个有用的函数，它可以将迭代器分组并返回一个子列表。以上代码演示了如何使用itertools库将一个大型列表分成多个较小的块。