将平面数组转换为多维数组

def flatten_to_multi_dimensional(arr, shape):
    if len(arr) != shape[0] * shape[1]:
        raise ValueError("The size of the input array does not match the specified shape.")
    
    multi_dim_arr = [[0] * shape[1] for _ in range(shape[0])]
    index = 0
    for i in range(shape[0]):
        for j in range(shape[1]):
            multi_dim_arr[i][j] = arr[index]
            index += 1
    
    return multi_dim_arr

# 示例用法
flat_arr = [1, 2, 3, 4, 5, 6]
shape = (2, 3)
multi_dim_arr = flatten_to_multi_dimensional(flat_arr, shape)
print(multi_dim_arr)