Load raster data in Python