# Writing raster dataset
def write_tif(path_tif, array, geotransform, geoprojection):

    # Checking the dimension of array.
    dim_array = array.shape
    if len(dim_array) > 2:
        depth = dim_array[2]
    else:
        depth = 1

    # Initializing TIF dataset driver
    driver = gdal.GetDriverByName("GTiff")

    # Creating actual dataset from TIF drivers.
    outdata = driver.Create(
        path_tif, size[0], size[1], depth, gdal.GDT_Float32)


    # sets same geotransform as input
    outdata.SetGeoTransform(geotransform)

    # sets same projection as input
    outdata.SetProjection(geoprojection)

    # Writing data to array
    for i in range(depth):
        try:
            arr = array[:, :, i]
        except:
            arr = array[:, :]
        outdata.GetRasterBand(i+1).WriteArray(arr)

    # If you want these values transparent
    outdata.GetRasterBand(1).SetNoDataValue(-9999)

    # saves to disk!!
    outdata.FlushCache()

    # Releasing memory
    outdata = None