📜  为列表中的每 2 个项目创建列表 - Python (1)

📅  最后修改于: 2023-12-03 15:06:17.766000             🧑  作者: Mango

为列表中的每 2 个项目创建列表 - Python

在Python中,我们可以使用列表切片来将一个大列表分割成若干个小列表,即每两个元素为一组进行分割。这在某些情况下比较有用,例如需要将数据集分割成一定长度的小片段进行处理。

以下是一个python函数用于将一个大列表分割成若干个小列表:

def divide_list(lst, size):
    """
    将列表分割成步长为 size 的子列表
    :param lst: List 列表
    :param size: int 步长
    :return: List[List] 子列表
    """
    return [lst[i:i+size] for i in range(0, len(lst), size)]

使用方式如下所示:

my_list = [1, 2, 3, 4, 5, 6, 7, 8]
new_list = divide_list(my_list, 2)
print(new_list)

输出结果:

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

如此便将原始列表分割成了4个长度为2的子列表。

除了使用列表切片,我们还可以使用for循环手动遍历列表进行分割,例如以下代码:

def divide_list(lst, size):
    """
    将列表分割成步长为 size 的子列表
    :param lst: List 列表
    :param size: int 步长
    :return: List[List] 子列表
    """
    new_list = []
    for i in range(0, len(lst), size):
        new_list.append(lst[i:i+size])
    return new_list

这种方法的输出结果完全一致,只是实现的方式略有不同。

无论是采用列表切片还是手动遍历方式,我们都可以有效地将一个长列表分割成若干个小列表,满足不同需求的数据处理需求。