mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
34 lines
810 B
Python
34 lines
810 B
Python
import logging
|
|
import numpy as np
|
|
import sys
|
|
import rasterio
|
|
from rasterio.features import rasterize
|
|
from rasterio.transform import IDENTITY
|
|
|
|
|
|
logging.basicConfig(stream=sys.stderr, level=logging.INFO)
|
|
logger = logging.getLogger('rasterize_geometry')
|
|
|
|
|
|
rows = cols = 10
|
|
|
|
geometry = {
|
|
'type': 'Polygon',
|
|
'coordinates': [[(2, 2), (2, 4.25), (4.25, 4.25), (4.25, 2), (2, 2)]]}
|
|
|
|
|
|
with rasterio.Env():
|
|
result = rasterize([geometry], out_shape=(rows, cols))
|
|
with rasterio.open(
|
|
"test.tif",
|
|
'w',
|
|
driver='GTiff',
|
|
width=cols,
|
|
height=rows,
|
|
count=1,
|
|
dtype=np.uint8,
|
|
nodata=0,
|
|
transform=IDENTITY,
|
|
crs={'init': "EPSG:4326"}) as out:
|
|
out.write(result.astype(np.uint8), indexes=1)
|