Python gdal获取tif影像的nodata区域为掩膜

要使用Python的GDAL库获取TIF影像的nodata区域并生成掩膜，你可以使用以下代码示例：

from osgeo import gdal
import numpy as np

# 打开影像文件
dataset = gdal.Open('image.tif')

# 获取影像的宽度和高度
width = dataset.RasterXSize
height = dataset.RasterYSize

# 创建一个与影像大小相同的布尔类型的掩膜
mask = np.zeros((height, width), dtype=bool)

# 读取影像数据
band = dataset.GetRasterBand(1)
data = band.ReadAsArray()

# 获取nodata值
nodata = band.GetNoDataValue()

# 根据nodata值生成掩膜
mask[data == nodata] = True

# 可以根据需要进行后续操作，比如保存掩膜为影像文件等

在上面的代码中，首先使用`gdal.Open()`函数打开TIF影像文件。然后，使用`RasterXSize`和`RasterYSize`属性获取影像的宽度和高度。接下来，创建一个与影像大小相同的布尔类型的掩膜，并使用`np.zeros()`函数初始化为全0。然后，使用`ReadAsArray()`方法读取影像数据，并使用`GetNoDataValue()`方法获取nodata值。最后，根据nodata值将掩膜中对应位置设置为True。

你可以根据需要进行后续操作，比如保存掩膜为影像文件等。