Lesson 1. Export Numpy Arrays to Geotiff Format Using Rasterio and Python

Export Raster Data with Rasterio - Earth analytics python course module

Welcome to the first lesson in the Export Raster Data with Rasterio module. In this module you will learn how to export raster data with rasterio in python.

Learning Objectives

After completing this tutorial, you will be able to:

  • Calculate NDVI using NAIP multispectral imagery in Python.
  • Describe what a vegetation index is and how it is used with spectral remote sensing data.
  • Export or write a raster to a .tif file from Python.

What You Need

You will need a computer with internet access to complete this lesson and the Cold Springs Fire data.

Download Week Cold Springs Fire Data (~150 MB)

Or use earthpy et.data.get_data('cold-springs-fire')

About Vegetation Indices

A vegetation index is a single value that quantifies vegetation health or structure. The math associated with calculating a vegetation index is derived from the physics of light reflection and absorption across bands. For instance, it is known that healthy vegetation reflects light strongly in the near infrared band and less strongly in the visible portion of the spectrum. Thus, if you create a ratio between light reflected in the near infrared and light reflected in the visible spectrum, it will represent areas that potentially have healthy vegetation.

Normalized Difference Vegetation Index (NDVI)

The Normalized Difference Vegetation Index (NDVI) is a quantitative index of greenness ranging from 0-1 where 0 represents minimal or no greenness and 1 represents maximum greenness.

NDVI is often used for a quantitate proxy measure of vegetation health, cover and phenology (life cycle stage) over large areas.

NDVI image from NASA that shows reflectance.
NDVI is calculated from the visible and near-infrared light reflected by vegetation. Healthy vegetation (left) absorbs most of the visible light that hits it, and reflects a large portion of near-infrared light. Unhealthy or sparse vegetation (right) reflects more visible light and less near-infrared light. Source: NASA

Calculate NDVI in Python

Sometimes you can download already calculated NDVI data products from a data provider.

However, in this case, you don’t have a pre calculated NDVI product from NAIP data. You need to calculate NDVI using the NAIP imagery / reflectance data that you have downloaded from Earth Explorer.

How to Derive the NDVI Vegetation Index From Multispectral Imagery

The normalized difference vegetation index (NDVI) uses a ratio between near infrared and red light within the electromagnetic spectrum. To calculate NDVI you use the following formula where NIR is near infrared light and red represents red light. For your raster data, you will take the reflectance value in the red and near infrared bands to calculate the index.

(NIR - Red) / (NIR + Red)

You can perform this calculation using matrix math with the numpy library.

To get started, load all of the required python libraries.

import rasterio as rio
import geopandas as gpd
import earthpy as et
import earthpy.spatial as es
import os
import matplotlib.pyplot as plt
from rasterio.plot import show
import numpy as np
# set working directory to your home dir/earth-analytics
os.chdir(os.path.join(et.io.HOME, 'earth-analytics'))
import matplotlib as mpl
mpl.rcParams['figure.figsize'] = (14, 14)
mpl.rcParams['axes.titlesize'] = 20

To begin open some data and create an output that you wish to export to geotiff format. Below you calculate NDVD from NAIP data.

with rio.open("data/cold-springs-fire/naip/m_3910505_nw_13_1_20150919/crop/m_3910505_nw_13_1_20150919_crop.tif") as src:
    naip_data = src.read()
naip_ndvi = (naip_data[3] - naip_data[0]) / (naip_data[3] + naip_data[0])
# need to add a colorbar to this image
fig, ax = plt.subplots(figsize=(12,6))
ndvi = ax.imshow(naip_ndvi, cmap='PiYG')
fig.colorbar(ndvi, fraction=.05)
ax.set(title="NAIP Derived NDVI\n 19 September 2015 - Cold Springs Fire, Colorado")
NDVI for the Cold Springs Fire site derived from NAIP data. You may want to export this data as a geotiff to share and use in other tools like QGIS.
NDVI for the Cold Springs Fire site derived from NAIP data. You may want to export this data as a geotiff to share and use in other tools like QGIS.

Export a Numpy Array to a Raster Geotiff in Python

When you are done, you can export your NDVI raster data so you could use them in QGIS or ArcGIS or share them with your colleagues. To do this you use the rio.write() function.

Exporting a raster in python is a bit different from what you may have learned using another language like R. In python, you need to:

  1. Create a new raster object with all of the metadata needed to define it. This metadata includes:
    • the shape (rows and columns) of the object
    • the coordinate reference system (crs)
    • the type of file (you will export a geotiff (.tif) in this lesson
    • and the type of data being stored (integer, float, etc).

Lucky for you, all of this information can be accessed from the original NAIP data that you imported into python using attribute calls like:

.transform and .crs

To implement this, below you will create a rasterio object (not using a context manager) to grab the needed spatial attributes.

with rio.open("data/cold-springs-fire/naip/m_3910505_nw_13_1_20150919/crop/m_3910505_nw_13_1_20150919_crop.tif") as src:
    naip_data_ras = src.read()
    naip_meta = src.profile
{'driver': 'GTiff', 'dtype': 'int16', 'nodata': -32768.0, 'width': 4377, 'height': 2312, 'count': 4, 'crs': CRS({'proj': 'utm', 'zone': 13, 'ellps': 'GRS80', 'towgs84': '0,0,0,0,0,0,0', 'units': 'm', 'no_defs': True}), 'transform': Affine(1.0, 0.0, 457163.0,
       0.0, -1.0, 4426952.0), 'tiled': False, 'compress': 'lzw', 'interleave': 'band'}
naip_transform = naip_meta["transform"]
naip_crs = naip_meta["crs"]
# View spatial attributes
naip_transform, naip_crs
(Affine(1.0, 0.0, 457163.0,
        0.0, -1.0, 4426952.0),
 CRS({'proj': 'utm', 'zone': 13, 'ellps': 'GRS80', 'towgs84': '0,0,0,0,0,0,0', 'units': 'm', 'no_defs': True}))

You can view the type of data stored within the ndvi array using .dtype. Remember that the naip_ndvi object is a numpy array.

type(naip_ndvi), naip_ndvi.dtype
(numpy.ndarray, dtype('float64'))

Use rio.open() to create a new blank raster ‘template’. Then write the NDVI numpy array to to that template using dst.write().

Note that when we write the data we need the following elements:

  1. the driver or type of file that we want to write. ‘Gtiff’ is a geotiff format
  2. dtype: the structure of the data that you are writing. We are writing floating point values (values with decimal places)
  3. the heigth and width of the ndvi object (accessed using the .shape attribute)
  4. the crs of the spatial object (accessed using the rasterio NAIP data)
  5. the transform information (accessed using the rasterio NAIP data)

Finally you need to specify the name of the output file and the path to where it will be saved on your computer.

Export a Numpy Array to a Raster Geotiff Using the Spatial Profile or Metadata of Another Raster

You can use the naip_meta variable that you created above. This variable contains all of the spatial metadata for naip data.

In this case, the

  1. number of bands (we have only one band vs 4 in the color image) and
  2. the data format (we have floating point numbers - numers with decimals - vs integers)

have changed. Update those values then write out the image.


{'driver': 'GTiff', 'dtype': 'int16', 'nodata': -32768.0, 'width': 4377, 'height': 2312, 'count': 4, 'crs': CRS({'proj': 'utm', 'zone': 13, 'ellps': 'GRS80', 'towgs84': '0,0,0,0,0,0,0', 'units': 'm', 'no_defs': True}), 'transform': Affine(1.0, 0.0, 457163.0,
       0.0, -1.0, 4426952.0), 'tiled': False, 'compress': 'lzw', 'interleave': 'band'}
# change the count or number of bands from 4 to 1
naip_meta['count'] = 1
# change the data type to float rather than integer
naip_meta['dtype'] = "float64"
{'driver': 'GTiff', 'dtype': 'float64', 'nodata': -32768.0, 'width': 4377, 'height': 2312, 'count': 1, 'crs': CRS({'proj': 'utm', 'zone': 13, 'ellps': 'GRS80', 'towgs84': '0,0,0,0,0,0,0', 'units': 'm', 'no_defs': True}), 'transform': Affine(1.0, 0.0, 457163.0,
       0.0, -1.0, 4426952.0), 'tiled': False, 'compress': 'lzw', 'interleave': 'band'}

Note below that when we write the raster, we use **naip_meta The two ** tells Python to unpack all of the values in the naip_meta object to use as arguments when writing the geotiff file. We already updated the elements that we needed to above (count and dtype). So this naip_meta object is ready to be used for the NDVI raster.

# write your the ndvi raster object
with rio.open('data/cold-springs-fire/outputs/naip_ndvi.tif', 'w', **naip_meta) as dst:
    dst.write(naip_ndvi, 1)

Leave a Comment