NumPy库为Python提供了高效的列表分割方法。本文将介绍两种使用NumPy高效分割列表的方案，适用于列表长度可被分割数量整除和无法整除的两种情况。
场景： 将一个包含30个元素的列表分割成多个子列表。
方法一：使用reshape()函数 (列表长度可被整除)
当列表长度能够被分割数量整除时，reshape()函数是最佳选择。它能将数组重塑成指定形状。以下代码演示如何将30个元素的列表分割成3个，每个包含10个元素的子列表：
立即学习“Python免费学习笔记（深入）”；
import numpy as np data = list(range(30)) # 创建一个包含0-29的列表 result = np.array(data).reshape((3, 10)) print(result)
这段代码先将列表转换为NumPy数组，然后使用reshape((3, 10))将其重塑为3行10列的二维数组，从而实现列表分割。
方法二：使用array_split()函数 (列表长度不可被整除)
如果列表长度不能被分割数量整除，reshape()函数将失效。此时，array_split()函数非常有用。它可以将数组分割成近似大小的子数组。以下代码演示了如何使用array_split()函数，即使列表长度无法被分割数量整除：
import numpy as np data = list(range(30)) result = np.array_split(data, 3) # 分割成3个子数组 print(result)
这段代码将列表分割成3个子数组，即使元素个数不能被3整除，array_split也会尽可能平均分配元素。
通过以上两种方法，您可以根据实际情况选择合适的NumPy函数来高效地分割列表，提升代码效率。 reshape()适用于整除情况，array_split()则适用于更通用的情况。
