将遮罩阵列指定给新光栅

Question

在Python中，我试图将通过将光栅文件屏蔽为光栅而获得的数组重新转换为光栅

更具体地说：我有两个光栅a）森林砍伐年份的数据光栅b）遮罩光栅，其中报告了映射的面积（=1）、零面积和水体（分别=0和=2）

我已经生成了一个掩码数组，其中我基于原始光栅替换了空值：

# Define relative paths
treeloss_path = os.path.join("/path/lossyear_congo.tif")
datamask_path = os.path.join("/path/datamask_congo.tif")`

# Open making layer
with rio.open(datamask_path) as datamask_cl:
    datamask = datamask_cl.read()
    datamask_ext = plotting_extent(datamask_cl)


# Open the baseman layer for your
with rio.open(treeloss_path) as treeloss_cl:
    treeloss = treeloss_cl.read()
    treeloss_ext = plotting_extent(treeloss_cl)`

然后我创建了我的蒙版数组，如下所示：

    import earthpy.mask as em
treecover_free_masked = em.mask_pixels(treeloss, datamask, vals=[0, 2])

现在，我需要创建一个具有遮罩阵列（treecover_free_masked）的光栅，该阵列在空间和原始数据的地理参考方面具有所有相同的功能。 有没有关于如何达到这一步的建议

谢谢