解开一维列表返回到三维数组

def convert_to_3d_array(lst, dim1, dim2, dim3):
    if dim1 * dim2 * dim3 != len(lst):
        return None  # 列表长度与三维数组维度不匹配，返回空值

    result = []
    index = 0
    for i in range(dim1):
        temp1 = []
        for j in range(dim2):
            temp2 = []
            for k in range(dim3):
                temp2.append(lst[index])
                index += 1
            temp1.append(temp2)
        result.append(temp1)

    return result

lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
dim1 = 2
dim2 = 2
dim3 = 3

result = convert_to_3d_array(lst, dim1, dim2, dim3)
print(result)

[[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]]