Python提取多边形NDVI均值方法详解

本文详细介绍了使用Python提取栅格图像中多边形区域内NDVI均值的方法，并扩展到提取多边形外部NDVI均值。针对遥感影像分析，例如Landsat影像的NDVI数据，通过结合`Rasterio`和`Fiona`库，实现了从Shapefile文件中读取多边形边界，并利用掩膜操作精准提取目标区域的NDVI值。文中提供了清晰的代码示例，展示了如何使用`rasterio.mask.mask()`函数，并针对初学者，详细解释了代码逻辑，包括异常处理和参数设置，如`invert=True`用于提取多边形外部的NDVI均值。此外，还介绍了同时提取多个多边形区域NDVI均值的方法，为土地覆盖分类、植被监测等遥感分析任务提供实用指南。关键词：Python，NDVI，Rasterio，Fiona，遥感影像，多边形，均值提取。

本文介绍如何使用Python提取栅格图像（例如Landsat5影像生成的NDVI图像）中，特定多边形区域内和区域外的NDVI均值。我们将使用Rasterio和Fiona这两个强大的库，通过读取Shapefile文件获取多边形边界，然后使用掩膜操作提取指定区域的NDVI值，最后计算均值。

准备工作

在开始之前，请确保已安装以下Python库：

• Rasterio: 用于读取和写入栅格数据。
• Fiona: 用于读取和写入矢量数据，例如Shapefile。
• Numpy: 用于数值计算，例如计算均值。

可以使用pip安装这些库：

pip install rasterio fiona numpy

提取多边形内部NDVI均值

以下代码演示了如何提取Shapefile文件定义的多边形内部的NDVI均值。

import rasterio
import fiona
import rasterio.mask
import numpy as np

# 定义输入文件路径
shapefile_path = "path/to/your/shapefile.shp" # 替换为你的Shapefile文件路径
raster_path = "path/to/your/ndvi.tif" # 替换为你的NDVI栅格文件路径

try:
    # 加载Shapefile文件，读取多边形几何信息
    with fiona.open(shapefile_path, "r") as sf:
        shapes = [feature["geometry"] for feature in sf]

    # 打开栅格文件
    with rasterio.open(raster_path) as src:
        # 使用掩膜提取多边形内部的像素值
        out_image, out_transform = rasterio.mask.mask(src, shapes, crop=True)

        # 计算NDVI均值
        NDVI_mean = np.mean(out_image)

        print(f"多边形内部NDVI均值: {NDVI_mean}")

except fiona.errors.DriverError as e:
    print(f"Fiona 错误: 无法打开 Shapefile 文件。请检查文件路径和文件格式。\n{e}")
except rasterio.RasterioIOError as e:
    print(f"Rasterio 错误: 无法打开栅格文件。请检查文件路径和文件格式。\n{e}")
except Exception as e:
    print(f"发生未知错误: {e}")

代码解释：

1. 导入必要的库: 导入 rasterio, fiona, rasterio.mask 和 numpy。
2. 定义文件路径: shapefile_path 变量存储Shapefile文件的路径，raster_path 变量存储NDVI栅格文件的路径。请根据实际情况修改这些路径。
3. 加载Shapefile文件: 使用 fiona.open() 函数打开Shapefile文件，并读取所有要素的几何信息。 shapes 变量是一个包含所有多边形几何对象的列表。
4. 打开栅格文件: 使用 rasterio.open() 函数打开NDVI栅格文件。
5. 使用掩膜提取像素值: rasterio.mask.mask() 函数使用Shapefile中的多边形作为掩膜，提取栅格图像中多边形内部的像素值。 crop=True 参数表示裁剪输出图像到掩膜的范围。 out_image 变量存储提取的像素值，out_transform 变量存储输出图像的地理变换信息。
6. 计算NDVI均值: 使用 numpy.mean() 函数计算提取的像素值的均值。
7. 打印结果: 打印计算得到的NDVI均值。
8. 异常处理: 使用 try...except 块来捕获可能发生的异常，例如文件无法打开或文件格式错误。

提取多边形外部NDVI均值

要提取多边形外部的NDVI均值，可以使用 rasterio.mask.mask() 函数的 invert=True 参数。

import rasterio
import fiona
import rasterio.mask
import numpy as np

# 定义输入文件路径
shapefile_path = "path/to/your/shapefile.shp" # 替换为你的Shapefile文件路径
raster_path = "path/to/your/ndvi.tif" # 替换为你的NDVI栅格文件路径

try:
    # 加载Shapefile文件，读取多边形几何信息
    with fiona.open(shapefile_path, "r") as sf:
        shapes = [feature["geometry"] for feature in sf]

    # 打开栅格文件
    with rasterio.open(raster_path) as src:
        # 使用掩膜提取多边形外部的像素值，设置 invert=True
        out_image, out_transform = rasterio.mask.mask(src, shapes, crop=True, invert=True)

        # 计算NDVI均值
        NDVI_mean = np.mean(out_image)

        print(f"多边形外部NDVI均值: {NDVI_mean}")

except fiona.errors.DriverError as e:
    print(f"Fiona 错误: 无法打开 Shapefile 文件。请检查文件路径和文件格式。\n{e}")
except rasterio.RasterioIOError as e:
    print(f"Rasterio 错误: 无法打开栅格文件。请检查文件路径和文件格式。\n{e}")
except Exception as e:
    print(f"发生未知错误: {e}")

注意：

• invert=True 参数指示 rasterio.mask.mask() 函数提取多边形 外部 的像素值。
• 确保Shapefile文件和栅格文件位于正确的路径，并且具有正确的格式。

同时提取多个多边形区域的NDVI均值

如果要提取多个多边形区域的NDVI均值，只需在Shapefile文件中包含多个要素（多边形）。 rasterio.mask.mask() 函数会自动处理Shapefile文件中的所有多边形。

总结

本文介绍了如何使用Python和Rasterio、Fiona库提取栅格图像中多边形区域内和区域外的NDVI均值。 这些技术可以应用于各种遥感分析任务，例如土地覆盖分类、植被监测和环境评估。 通过灵活运用 rasterio.mask.mask() 函数，可以方便地提取指定区域的栅格数据，并进行进一步的分析。

今天关于《Python提取多边形NDVI均值方法详解》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！