Rasterizing by attribute in rasterio

Question

According to Rasterizing shapefiles with GDAL and Python? it seems that gdal allows rasterization of a layer according to a certain attribute

ds = gdal.RasterizeLayer(target_ds, [1], source_layer, options = ["ATTRIBUTE=ID"])

I would like to rasterize a layer with small square polygon geometries using rasterio

rasterio has a features.rasterize method (see docs), but it seems to only be able to output numpy arrays with 1's where the polygons occur. Since rasterio is a higher level library than gdal I am a bit surprised that selecting an attribute to burn the raster with is not supported functionality.

Is there any built-in method in rasterio to burn a raster by an attribute from the original vector layer?

Jon · Accepted Answer

The inputs to features.rasterize include

shapes (iterable of (geometry, value) pairs or iterable over) –
  geometries. geometry can either be an object that implements the geo
  interface or GeoJSON-like object.

So you can just pass tuples of (geom, value), where value is the attribute value you wish to burn into the raster.

mmann1123 · Answer

To give an example if you have col set to the column you want to rasterize
if col:
    shapes = ((geom,value) for geom, value in zip(dataframe.geometry, dataframe[col]))
else: 
    shapes = dataframe.geometry.values
    
varray = rasterize(shapes,
                   out_shape=(dst_height, dst_width),
                   transform=dst_transform,
                   fill=fill,
                   default_value=default_value,
                   all_touched=all_touched,
                   dtype=dtype)

Rasterizing by attribute in rasterio

2 Answers

Add your own answers!

Ask a Question