rasterio/examples/rasterize_geometry.py
2014-03-28 15:42:35 -07:00

29 lines
800 B
Python

import logging
import numpy
import sys
import rasterio
from rasterio.features import rasterize_geometries
logging.basicConfig(stream=sys.stderr, level=logging.INFO)
logger = logging.getLogger('rasterize_geometry')
rows = cols = 10
transform = [0, 1, 0, 0, 0, 1]
geometry = {'type':'Polygon','coordinates':[[(2,2),(2,4.25),(4.25,4.25),(4.25,2),(2,2)]]}
with rasterio.drivers():
result = rasterize_geometries([geometry], rows, cols, transform)
with rasterio.open(
"test.tif",
'w',
driver='GTiff',
width=cols,
height=rows,
count=1,
dtype=numpy.uint8,
nodata=0,
transform=transform,
crs={'init': "EPSG:4326"}) as out:
out.write_band(1, result.astype(numpy.uint8))