Intersection joiner
In [1]:
Copied!
import geopandas as gpd
from shapely import geometry
from srai.constants import FEATURES_INDEX, REGIONS_INDEX, WGS84_CRS
from srai.plotting.folium_wrapper import plot_regions
import geopandas as gpd
from shapely import geometry
from srai.constants import FEATURES_INDEX, REGIONS_INDEX, WGS84_CRS
from srai.plotting.folium_wrapper import plot_regions
Define example regions and features¶
In [2]:
Copied!
regions = gpd.GeoDataFrame(
geometry=[
geometry.Polygon([(-1, 0), (-1, -1), (0, -1), (0, 0)]),
geometry.Polygon([(1, 0), (1, 1), (0, 1), (0, 0)]),
geometry.Polygon([(-2, -1), (-2, -2), (-1, -2), (-1, -1)]),
geometry.Polygon([(-2, 0.5), (-2, -0.5), (-1, -0.5), (-1, 0.5)]),
],
crs=WGS84_CRS,
index=gpd.pd.Index(name=REGIONS_INDEX, data=[1, 2, 3, 4]),
)
features = gpd.GeoDataFrame(
geometry=[
geometry.Polygon([(-1.5, 0.5), (-1.5, 0), (-0.5, 0), (-0.5, 0.5)]),
geometry.Polygon([(-1.5, -1.5), (-1.5, -2.5), (-0.5, -2.5), (-0.5, -1.5)]),
geometry.Point((0, 0)),
geometry.Point((-0.5, -0.5)),
],
crs=WGS84_CRS,
index=gpd.pd.Index(name=FEATURES_INDEX, data=[1, 2, 3, 4]),
)
regions = gpd.GeoDataFrame(
geometry=[
geometry.Polygon([(-1, 0), (-1, -1), (0, -1), (0, 0)]),
geometry.Polygon([(1, 0), (1, 1), (0, 1), (0, 0)]),
geometry.Polygon([(-2, -1), (-2, -2), (-1, -2), (-1, -1)]),
geometry.Polygon([(-2, 0.5), (-2, -0.5), (-1, -0.5), (-1, 0.5)]),
],
crs=WGS84_CRS,
index=gpd.pd.Index(name=REGIONS_INDEX, data=[1, 2, 3, 4]),
)
features = gpd.GeoDataFrame(
geometry=[
geometry.Polygon([(-1.5, 0.5), (-1.5, 0), (-0.5, 0), (-0.5, 0.5)]),
geometry.Polygon([(-1.5, -1.5), (-1.5, -2.5), (-0.5, -2.5), (-0.5, -1.5)]),
geometry.Point((0, 0)),
geometry.Point((-0.5, -0.5)),
],
crs=WGS84_CRS,
index=gpd.pd.Index(name=FEATURES_INDEX, data=[1, 2, 3, 4]),
)
In [3]:
Copied!
regions
regions
Out[3]:
| geometry | |
|---|---|
| region_id | |
| 1 | POLYGON ((-1 0, -1 -1, 0 -1, 0 0, -1 0)) |
| 2 | POLYGON ((1 0, 1 1, 0 1, 0 0, 1 0)) |
| 3 | POLYGON ((-2 -1, -2 -2, -1 -2, -1 -1, -2 -1)) |
| 4 | POLYGON ((-2 0.5, -2 -0.5, -1 -0.5, -1 0.5, -2... |
In [4]:
Copied!
features
features
Out[4]:
| geometry | |
|---|---|
| feature_id | |
| 1 | POLYGON ((-1.5 0.5, -1.5 0, -0.5 0, -0.5 0.5, ... |
| 2 | POLYGON ((-1.5 -1.5, -1.5 -2.5, -0.5 -2.5, -0.... |
| 3 | POINT (0 0) |
| 4 | POINT (-0.5 -0.5) |
In [5]:
Copied!
folium_map = plot_regions(regions, colormap=["royalblue"])
features.explore(
m=folium_map,
style_kwds=dict(color="red", opacity=0.8, fillColor="red", fillOpacity=0.5),
marker_kwds=dict(radius=3),
)
folium_map = plot_regions(regions, colormap=["royalblue"])
features.explore(
m=folium_map,
style_kwds=dict(color="red", opacity=0.8, fillColor="red", fillOpacity=0.5),
marker_kwds=dict(radius=3),
)
Out[5]:
Make this Notebook Trusted to load map: File -> Trust Notebook
Join regions with features using intersection joiner¶
In [6]:
Copied!
from srai.joiners import IntersectionJoiner
joiner = IntersectionJoiner()
joint = joiner.transform(regions, features, return_geom=True)
joint
from srai.joiners import IntersectionJoiner
joiner = IntersectionJoiner()
joint = joiner.transform(regions, features, return_geom=True)
joint
Out[6]:
| geometry | ||
|---|---|---|
| region_id | feature_id | |
| 1 | 3 | POINT (0 0) |
| 2 | 3 | POINT (0 0) |
| 1 | 4 | POINT (-0.5 -0.5) |
| 1 | LINESTRING (-1 0, -0.5 0) | |
| 4 | 1 | POLYGON ((-1.5 0, -1.5 0.5, -1 0.5, -1 0, -1.5... |
| 3 | 2 | POLYGON ((-1.5 -1.5, -1 -1.5, -1 -2, -1.5 -2, ... |
In [7]:
Copied!
folium_map = plot_regions(regions, colormap=["royalblue"])
features.explore(
m=folium_map,
style_kwds=dict(color="red", opacity=0.5, fillColor="red", fillOpacity=0.5),
marker_kwds=dict(radius=3),
)
joint.explore(
m=folium_map,
style_kwds=dict(color="yellow", opacity=1.0, fillColor="yellow", fillOpacity=1.0),
marker_kwds=dict(radius=3),
)
folium_map = plot_regions(regions, colormap=["royalblue"])
features.explore(
m=folium_map,
style_kwds=dict(color="red", opacity=0.5, fillColor="red", fillOpacity=0.5),
marker_kwds=dict(radius=3),
)
joint.explore(
m=folium_map,
style_kwds=dict(color="yellow", opacity=1.0, fillColor="yellow", fillOpacity=1.0),
marker_kwds=dict(radius=3),
)
Out[7]:
Make this Notebook Trusted to load map: File -> Trust Notebook