access underlying rasterio or gdal dataset?

[cefect]

Is there a method or convenient way for extracting (or creating) a gdal.Dataset or rasterio dataset? For example, something to accomplish the following:

    # Create in-memory source raster
    drv = gdal.GetDriverByName("MEM")  
    src_ds = drv.Create("", dataarray.shape[1], dataarray.shape[0], 1, gdal.GDT_Float32)
    src_band = src_ds.GetRasterBand(1)
    src_band.WriteArray(dataarray.data)
    src_band.SetNoDataValue(int(dataarray.rio.nodata) or -9999)  # Set nodata from DataArray
 
    # Set GeoTransform and Projection
    # Extract GeoTransform from Affine object
    transform = dataarray.rio.transform()
    geotransform = (transform.a, transform.b, transform.c, 
                    transform.d, transform.e, transform.f)
     
    src_ds.SetGeoTransform(geotransform)
    src_ds.SetProjection(dataarray.rio.crs.to_wkt())

[snowman2]

With rioxarray 0.21+ (#894):

with dataarray.rio.to_rasterio_dataset() as rio_ds:
    ...

With older rioxarray versions:

from rasterio.io import MemoryFile

with MemoryFile() as memfile:
    dataarray.rio.to_raster(memfile.name)
    with memfile.open() as src_ds:
        ....