如何在python中将数组划分为多个子数组

def chunk_list(lst, n):
    """Yield successive n-sized chunks from lst."""
    for i in range(0, len(lst), n):
        yield lst[i:i + n]

# 示例数组
array = [1, 2, 3, 4, 5, 6, 7, 8, 9]

# 划分为大小为3的子数组
sub_arrays = list(chunk_list(array, 3))
print(sub_arrays)  # 输出: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

def split_by_condition(lst, condition_func):
    sub_arrays = []
    current_sub_array = []

    for item in lst:
        if condition_func(item):
            if current_sub_array:
                sub_arrays.append(current_sub_array)
                current_sub_array = []
        else:
            current_sub_array.append(item)

    if current_sub_array:
        sub_arrays.append(current_sub_array)

    return sub_arrays

# 示例数组
array = [1, 2, 3, 4, 5, 6, 7, 8, 9]

# 划分条件：偶数开头
sub_arrays = split_by_condition(array, lambda x: x % 2 == 0)
print(sub_arrays)  # 输出: [[1], [2, 3], [4], [5, 6], [7], [8, 9]]

import numpy as np

# 示例数组
array = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])

# 划分为大小为3的子数组
sub_arrays = np.array_split(array, 3)
print(sub_arrays)  # 输出: [array([1, 2, 3]), array([4, 5, 6]), array([7, 8, 9])]