本文档旨在提供一个使用 Python 从栅格图像（例如 NDVI 图像）中的多边形区域提取统计值的实用指南。我们将使用 rasterio 和 fiona 库来读取栅格数据和矢量数据，并使用 rasterio.mask 来裁剪栅格数据。最终，我们将计算裁剪后区域的 NDVI 平均值。本教程适用于需要对地理空间数据进行分析和处理的专业人士和研究人员。
从栅格图像中提取多边形区域的 NDVI 值
本教程将介绍如何使用 Python 从栅格图像（例如 NDVI 图像）中的多边形区域提取 NDVI 值。我们将使用 rasterio 和 fiona 库来实现这个目标。
准备工作
在开始之前，请确保已安装以下 Python 库：
- rasterio: 用于读取和写入栅格数据。
- fiona: 用于读取和写入矢量数据。
- numpy: 用于数值计算。
可以使用 pip 安装这些库：
立即学习“Python免费学习笔记（深入）”；
pip install rasterio fiona numpy
步骤
以下是提取 NDVI 值的步骤：
- 导入必要的库:
import rasterio import fiona import rasterio.mask import numpy as np
- 加载矢量数据 (Shapefile):
使用 fiona 库打开包含多边形几何信息的 shapefile。
path_to_shapefile = "path/to/your/shapefile.shp" # 替换为你的 shapefile 路径
with fiona.open(path_to_shapefile, "r") as sf:
shapes = [feature["geometry"] for feature in sf]shapes 变量现在包含 shapefile 中所有要素的几何信息。
- 加载栅格数据 (TIFF):
使用 rasterio 库打开包含 NDVI 数据的 TIFF 图像。
source_tif = "path/to/your/ndvi.tif" # 替换为你的 NDVI TIFF 图像路径
with rasterio.open(source_tif) as src:
pass #后续操作在with语句块内进行- 使用矢量数据裁剪栅格数据:
使用 rasterio.mask.mask 函数，根据多边形几何信息裁剪栅格数据。
with rasterio.open(source_tif) as src:
out_image, out_transform = rasterio.mask.mask(src, shapes, crop=True)out_image 变量现在包含裁剪后的栅格数据，out_transform 变量包含新的地理变换信息。 crop=True 参数确保裁剪后的图像只包含多边形范围内的像素。
- 计算 NDVI 平均值:
使用 numpy 库计算裁剪后的栅格数据的平均值。
NDVI_mean = np.mean(out_image)
print(f"NDVI 平均值: {NDVI_mean}")NDVI_mean 变量现在包含多边形区域内的 NDVI 平均值。
完整代码示例
import rasterio
import fiona
import rasterio.mask
import numpy as np
# 定义文件路径
path_to_shapefile = "path/to/your/shapefile.shp" # 替换为你的 shapefile 路径
source_tif = "path/to/your/ndvi.tif" # 替换为你的 NDVI TIFF 图像路径
# 加载边界
with fiona.open(path_to_shapefile, "r") as sf:
shapes = [feature["geometry"] for feature in sf]
# 裁剪栅格数据
with rasterio.open(source_tif) as src:
out_image, out_transform = rasterio.mask.mask(src, shapes, crop=True, filled=False) # filled=False ensures no data values are preserved
# 计算平均值
NDVI_mean = np.mean(out_image)
print(f"NDVI 平均值: {NDVI_mean}")从多边形外部提取数据
如果需要提取多边形外部的 NDVI 值，可以使用 rasterio.mask.mask 函数的 invert=True 参数。
with rasterio.open(source_tif) as src:
out_image, out_transform = rasterio.mask.mask(src, shapes, crop=True, invert=True, filled=False)invert=True 参数指示函数返回多边形外部的像素。filled=False 避免了多边形内部被填充为0。
注意事项
- 确保 shapefile 和 TIFF 图像的坐标系一致。如果坐标系不一致，需要进行坐标系转换。
- 如果 shapefile 中包含多个多边形，rasterio.mask.mask 函数会将所有多边形合并为一个多边形进行裁剪。如果需要分别提取每个多边形的 NDVI 值，需要循环遍历 shapefile 中的每个要素，并分别进行裁剪和计算。
- rasterio.mask.mask 函数返回的 out_image 是一个 NumPy 数组。可以使用 NumPy 的其他函数进行更复杂的统计分析。
- 如果栅格数据包含 NoData 值，在计算平均值之前需要将 NoData 值排除。可以使用 numpy.ma.masked_invalid 函数将 NoData 值屏蔽。
总结
本教程介绍了如何使用 Python 从栅格图像中的多边形区域提取 NDVI 值。通过使用 rasterio 和 fiona 库，可以方便地读取和处理地理空间数据，并进行统计分析。希望本教程对您有所帮助！
