Python与GDAL实现NDVI计算的实践与优化

Hazel

发布时间：2026-10-13| 6人浏览过

原创

本文介绍了使用Python和GDAL库计算归一化植被指数（NDVI）的方法。针对数组除法中常见的整数运算陷阱，通过将波段数据转换为浮点型确保结果范围在-1到1之间。同时对比了纯Python处理与C++结合GDAL的计算效率，指出后者在速率上的显著优势，为遥感数据处理提供实用参考。

NDVI计算中的数组运算问题

虽然代码篇幅简短，但实际编写过程耗时较长。主要难点在于数组除法运算的结果异常，通常NDVI值应介于-1至1之间。参考其他资料中的NDVI计算代码时，发现可能存在某些优化函数导致结果错误，因此决定放弃原有方案，重新调整思路。

数据类型转换与结果验证

采用直接计算的方法，关键在于处理影像波段的数据类型。由于原始波段数据为整数，需先将其转换为浮点型数值，再参与后续运算。经过类型转换后，计算得出的NDVI结果范围符合预期，验证了该方法的正确性。

尽管上述Python方案可行，但计算效率仍有提升空间。相比之下，使用C++结合GDAL库进行计算，能够显著提升处理速率，更适合大规模遥感影像的处理需求。

Python与C++性能对比


from osgeo import gdal_array as ga
import gdal, ogr, os, osr
import numpy as np
b3=r'C:\Users\suns\Desktop\b4.TIF'
b4=r'C:\Users\suns\Desktop\b5.TIF'
arr=ga.LoadFile(b3)
arr1=ga.LoadFile(b4)
ga.numpy.seterr(all="ignore")
ndvi=((arr1-arr)*1.0)/((arr1+arr)*1.0)
ndvi1=ga.numpy.nan_to_num(ndvi)
target=r'C:\Users\suns\Desktop\ndvi1.tif'
out=ga.SaveArray(ndvi1,target,format = "GTiff",prototype = b4)
out=None