将2D数组重塑为3D数组以进行tiff转换

def reshape_2d_to_3d(input_array, rows, cols, depth):
    if rows * cols * depth != len(input_array) * len(input_array[0]):
        return "无法重塑为指定的3D数组大小"

    reshaped_array = [[[0] * depth for _ in range(cols)] for _ in range(rows)]
    r, c, d = 0, 0, 0

    for i in range(len(input_array)):
        for j in range(len(input_array[0])):
            reshaped_array[r][c][d] = input_array[i][j]
            d += 1
            if d == depth:
                d = 0
                c += 1
                if c == cols:
                    c = 0
                    r += 1

    return reshaped_array