Skip to content

Instantly share code, notes, and snippets.

@jgomezdans

jgomezdans/calc_ndvi_gdal.py

Created February 24, 2016 16:23
Show Gist options
  • Star 6 You must be signed in to star a gist
  • Fork 1 You must be signed in to fork a gist
  • Save jgomezdans/6c11fef62eced9e131eb to your computer and use it in GitHub Desktop.
Save jgomezdans/6c11fef62eced9e131eb to your computer and use it in GitHub Desktop.
Download ZIP
Simple script to calculate NDVI using pure GDAL
Raw
calc_ndvi_gdal.py
from osgeo import gdal
import numpy
g = gdal.Open ("baikal_subset.tif")
if g is None:
raise IOError, "Couldn't open baikal_subset.tif"
b3 = g.GetRasterBand(3).ReadAsArray().astype(np.float32)
b4 = g.GetRasterBand(4).ReadAsArray().astype(np.float32)
ndvi = (b4 - b3)/(b4 + b3)
drv = gdal.GetDriverByName ( "GTiff" )
dst_ds = drv.Create ( "output_file.tif", g.RasterXSize, g.RasterYSize, 1,
gdal.GDT_Float32, options=["COMPRESS=LZW"] )
dst_ds.GetRasterBand(1).WriteArray ( dst_ds.astype (np.float32) )
dst_ds = None
@GISJohnECS
Copy link

Quick and easy thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment