python代码。将数组拆分成2个，前n行一个数组，其他行一个数组

以下是python代码，它可以把数组分成两个：第1到n行的数据存在一个数组中，其余的数据存在另一个数组中。

def split_array(array, n):
    return array[:n], array[n:]

使用方法：

>>> array = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> n = 5
>>> part1, part2 = split_array(array, n)
>>> print(part1)
[1, 2, 3, 4, 5]
>>> print(part2)
[6, 7, 8, 9, 10]