""" Rasterize shapefiles ====================== Importing the necessary packages: """ import matplotlib.pyplot as plt import imod # %% # Get the example shapes temp_dir = imod.util.temporary_directory() lakes = imod.data.lakes_shp(temp_dir) # %% # We'll need a dummy grid which we will use as a reference for for rasterizing # the shapefile. We are just going to create one with the convenience function. xmin = 90950.0 xmax = 115650.0 dx = 100 ymin = 445850.0 ymax = 467550.0 dy = -100.0 like_2d = imod.util.empty_2d(dx, xmin, xmax, dy, ymin, ymax) # %% # Rasterrize the shapes lake_grid = imod.prepare.rasterize(lakes, like=like_2d) # Plot fig, ax = plt.subplots() lake_grid.plot.imshow(ax=ax) # %% # To rasterize on a different grid, create a dummy grid dx_coarse = 200 dy_coarse = -200 like_2d_coarse = imod.util.empty_2d(dx_coarse, xmin, xmax, dy_coarse, ymin, ymax) lake_grid_coarse = imod.prepare.rasterize(lakes, like=like_2d_coarse) # Plot fig, ax = plt.subplots() lake_grid_coarse.plot.imshow(ax=ax)