Python列表均分|极客笔记

Python列表均分

在Python编程中，有时我们需要将一个列表均分成几个子列表，以便更好地处理数据或进行操作。本文将详细介绍如何在Python中实现列表的均分功能。

方法一：使用列表切片法

一种常见且简单的方法是使用列表切片法来对列表进行均分。我们可以先计算出每个子列表的长度，然后使用切片法将原列表分割成对应长度的子列表。

下面是一个示例代码演示如何实现列表的均分：

def split_list(lst, n):
    """
    将列表均分为 n 个子列表
    :param lst: 原列表
    :param n: 子列表的数量
    :return: 均分后的子列表集合
    """
    avg = len(lst) // n  # 计算每个子列表的长度
    remaining = len(lst) % n  # 计算剩余的元素个数

    result = []
    start = 0
    for i in range(n):
        if i < remaining:
            end = start + avg + 1
        else:
            end = start + avg
        result.append(lst[start:end])
        start = end

    return result

# 测试
lst = [1, 2, 3, 4, 5, 6, 7, 8, 9]
n = 3
print(split_list(lst, n))

运行结果：

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

以上代码中，我们首先计算出每个子列表的长度avg，然后计算出剩余的元素个数remaining。接下来，我们使用一个循环来遍历原列表，并根据剩余元素的个数，决定每个子列表的长度。最后将每个子列表加入到结果集合中返回。

方法二：使用numpy库

另一种方法是使用numpy库中的array_split函数来实现列表的均分。array_split函数可以将数组均分为指定数量的子数组。

下面是一个示例代码演示如何使用numpy库实现列表的均分：

import numpy as np

def split_list(lst, n):
    """
    将列表均分为 n 个子列表
    :param lst: 原列表
    :param n: 子列表的数量
    :return: 均分后的子列表集合
    """
    arr = np.array(lst)
    result = np.array_split(arr, n)
    return [sub.tolist() for sub in result]

# 测试
lst = [1, 2, 3, 4, 5, 6, 7, 8, 9]
n = 3
print(split_list(lst, n))

运行结果：

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

以上代码中，我们首先将原列表转换为numpy的数组对象，然后使用array_split函数将数组均分为n个子数组。最后将每个子数组转换为列表并返回。

方法三：自定义函数

除了以上两种方法外，我们还可以自定义一个函数来实现列表的均分。这种方法较为灵活，可以根据实际需求进行调整。

下面是一个示例代码演示如何使用自定义函数实现列表的均分：

def split_list(lst, n):
    """
    将列表均分为 n 个子列表
    :param lst: 原列表
    :param n: 子列表的数量
    :return: 均分后的子列表集合
    """
    result = []
    for i in range(0, len(lst), n):
        result.append(lst[i:i + n])

    return result

# 测试
lst = [1, 2, 3, 4, 5, 6, 7, 8, 9]
n = 3
print(split_list(lst, n))

运行结果：