将一个二维数组的子集分配给另一个不同大小的数组

def allocate_subset(source_array, target_array):
    # 获取源数组和目标数组的大小和维度
    source_rows, source_cols = len(source_array), len(source_array[0])
    target_rows, target_cols = len(target_array), len(target_array[0])

    # 选择子集（示例中选择源数组的前两行和前两列作为子集）
    subset = [row[:2] for row in source_array[:2]]

    # 分配子集到目标数组中
    for i in range(min(source_rows, target_rows)):
        for j in range(min(source_cols, target_cols)):
            target_array[i][j] = subset[i][j]

    return target_array

# 示例用法
source_array = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]
target_array = [[0, 0, 0], [0, 0, 0]]

result = allocate_subset(source_array, target_array)
print(result)