Lesson 3. Clip a spatial vector layer in Python using Shapely & GeoPandas: GIS in Python
Learning Objectives
- Clip a spatial vector point and line layer to the spatial extent of a polygon layer in Python using geopandas.
- Plot data with custom legends.
How to Clip Vector Data in Python
What Is Clipping or Cropping Data?
When you clip or crop spatial data you are removing the data outside of the clip extent. This means that your clipped dataset will be SMALLER (have a smaller spatial extent) than the original dataset. This also means that objects in the data such as polygons or lines will be CUT based on the boundary of the clip object.
When Do You Want to Clip Data?
You may want to clip your data for several reasons:
- You have more data than you need. For example you have data outside of your study area that you don’t need to process. Clipping it to the study area boundary will make it smaller and easier to manage!
- If you have data outside of your study area and you clip it, you can perform analysis on only that region - thus you won’t need to subset the data further when you perform summary statistics for example.
- When you plot the data you will only see the study region.
You will learn how to both crop your data and zoom in on an extent below.
Get started by loading your libraries. And be sure that your working directory is set.
In this lesson you will find examples of how to clip point and line data using geopandas. At the bottom of the lesson you will see a set of functions that can be used to clip the data in just one line of code. This lesson explains how those functions are built.
# Import libraries
import os
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.lines as mlines
from matplotlib.colors import ListedColormap
import geopandas as gpd
# Load the box module from shapely to create box objects
from shapely.geometry import box
import earthpy as et
import seaborn as sns
# Ignore warning about missing/empty geometries
import warnings
warnings.filterwarnings('ignore', 'GeoSeries.notna', UserWarning)
# Adjust plot font sizes
sns.set(font_scale=1.5)
sns.set_style("white")
# Set working dir & get data
data = et.data.get_data('spatial-vector-lidar')
os.chdir(os.path.join(et.io.HOME, 'earth-analytics'))
Downloading from https://ndownloader.figshare.com/files/12459464
Extracted output to /root/earth-analytics/data/spatial-vector-lidar/.
How to Clip Shapefiles in Python
In your dataset for this week you have 3 layers:
- A country boundary for the USA and
- A state boundary layer for the USA and
- Populated places in the USA
The data are imported and plotted below. Notice that there are points outside of your study area which is the continental USA. Your goal is the clip the points out that you NEED for your project - the points that overlay on top of the continental United States.
# Import all of your data at the top of your notebook to keep things organized.
country_boundary_us_path = os.path.join("data", "spatial-vector-lidar",
"usa", "usa-boundary-dissolved.shp")
country_boundary_us = gpd.read_file(country_boundary_us_path)
state_boundary_us_path = os.path.join("data", "spatial-vector-lidar",
"usa", "usa-states-census-2014.shp")
state_boundary_us = gpd.read_file(state_boundary_us_path)
pop_places_path = os.path.join("data", "spatial-vector-lidar", "global",
"ne_110m_populated_places_simple", "ne_110m_populated_places_simple.shp")
pop_places = gpd.read_file(pop_places_path)
# Are the data all in the same crs?
print("country_boundary_us", country_boundary_us.crs)
print("state_boundary_us", state_boundary_us.crs)
print("pop_places", pop_places.crs)
country_boundary_us epsg:4326
state_boundary_us epsg:4326
pop_places epsg:4326
# Plot the data
fig, ax = plt.subplots(figsize=(12, 8))
country_boundary_us.plot(alpha=.5,
ax=ax)
state_boundary_us.plot(cmap='Greys',
ax=ax,
alpha=.5)
pop_places.plot(ax=ax)
plt.axis('equal')
ax.set_axis_off()
plt.show()
Clip The Points Shapefile in Python Using Geopandas
To remove the points that are outside of your study area, you can clip the data. Removing or clipping data can make the data smaller and inturn plotting and analysis faster.
To clip points, lines, and polygons, GeoPandas has a function named clip()
that will clip all types of geometries. This operation used to be much more difficult, involving creating bounding boxes and shapely objects, while using the GeoPandas intersection()
function to clip the data. However, to simplify the process EarthPy developed a clip_shp()
function that would do all of these things automatically. The function was than picked up by GeoPandas and is now a part of their package! Just a small example of how awesome working with open source software can be.
Clip()
takes three arguments:
- gdf: Vector layer (point, line, polygon) to be clipped to mask.
- mask: Polygon vector layer used to clip
gdf
. The mask’s geometry is dissolved into one geometric feature and intersected withgdf
. - keep_geom_type: If True, return only geometries of original type in case of intersection resulting in multiple geometry types or GeometryCollections. If False, return all resulting geometries (potentially mixed-types). Default value is False (You don’t need to worry about this argument for this assignment)
clip()
will clip the data to the boundary of the polygon layer that you select. If there are multiple polygons in your clip_obj object, clip()
will clip the data to the total boundary of all polygons in the layer.
# Clip the data using GeoPandas clip
points_clip = gpd.clip(pop_places, country_boundary_us)
# View the first 6 rows and a few select columns
points_clip[['name', 'geometry', 'scalerank', 'natscale', ]].head()
name | geometry | scalerank | natscale | |
---|---|---|---|---|
175 | San Francisco | POINT (-122.41717 37.76920) | 1 | 300 |
176 | Denver | POINT (-104.98596 39.74113) | 1 | 300 |
177 | Houston | POINT (-95.34193 29.82192) | 1 | 300 |
178 | Miami | POINT (-80.22605 25.78956) | 1 | 300 |
179 | Atlanta | POINT (-84.40190 33.83196) | 1 | 300 |
Now you can plot the data to see the newly “clipped” points layer.
# Plot the data
fig, ax = plt.subplots(figsize=(12, 8))
country_boundary_us.plot(alpha=1,
color="white",
edgecolor="black",
ax=ax)
state_boundary_us.plot(cmap='Greys',
ax=ax,
alpha=.5)
points_clip.plot(ax=ax,
column='name')
ax.set_axis_off()
plt.axis('equal')
# Label each point - note this is just shown here optionally but is not required for your homework
points_clip.apply(lambda x: ax.annotate(s=x['name'],
xy=x.geometry.coords[0],
xytext=(6, 6), textcoords="offset points",
backgroundcolor="white"),
axis=1)
plt.show()
<ipython-input-6-388fbc261ae3>:19: MatplotlibDeprecationWarning: The 's' parameter of annotate() has been renamed 'text' since Matplotlib 3.3; support for the old name will be dropped two minor releases later.
points_clip.apply(lambda x: ax.annotate(s=x['name'],
Clip a Line or Polygon Layer to An Extent
The process for clipping a line or polygon layer is slightly different than clipping a set of points. To clip a line of polygon feature you will do the following:
- Ensure that your polygon and line layer are in the same coordinate reference system
- Identify what features in the lines layer fall WITHIN the boundary of the polygon layer
- Subset the features within the geometry and reset the geometry of the newly clipped layer to be equal to the clipped data.
This last step may seem unusual. When you clip data using shapely and geopandas the default behaviour is for it to only return the clipped geometry. However you may wish to also retain the attributes associated with the geometry. This is where the set_geometry()
method comes into play.
For this example you will use the country_boundary
layer and a clipped version of the natural earth 10m roads layer. * Import ne_10m_n_america_roads.shp
into python.
- Next, check to ensure that the roads and country boundary are in the same CRS. You may need to reproject the data.
- Because spatial operations take time, it’s best if you subset your data as much as possible prior to clipping.
# Open the roads layer
ne_roads_path = os.path.join("data", "spatial-vector-lidar", "global",
"ne_10m_roads", "ne_10m_roads.shp")
ne_roads = gpd.read_file(ne_roads_path)
# Are both layers in the same CRS?
if (ne_roads.crs == country_boundary_us.crs):
print("Both layers are in the same crs!",
ne_roads.crs, country_boundary_us.crs)
Both layers are in the same crs! epsg:4326 epsg:4326
How to Clip Lines and Polygons in Python
In your dataset for this week you have 2 layers.
- A global, natural earth roads layer and
- A boundary for the United States.
The roads data are imported below. You imported the boundary layer above.
If both layers are in the same CRS, you are ready to clip your data. Because the clip functions are new and little testing has been done with them, you will see all of the lines of code required to clip your data. However you can use the clip_shp()
function to clip your data for this week’s class!
Be patient while your clip code below runs.
To make your code run faster, you can simplify the geometry of your country boundary. Simplify should be used with caution as it does modify your data.
# Simplify the geometry of the clip extent for faster processing
# Use this with caution as it modifies your data.
country_boundary_us_sim = country_boundary_us.simplify(
.2, preserve_topology=True)
Clip and plot the data.
# Clip data
ne_roads_clip = gpd.clip(ne_roads, country_boundary_us_sim)
# Ignore missing/empty geometries
ne_roads_clip = ne_roads_clip[~ne_roads_clip.is_empty]
print("The clipped data have fewer line objects (represented by rows):",
ne_roads_clip.shape, ne_roads.shape)
The clipped data have fewer line objects (represented by rows): (7346, 32) (56601, 32)
fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(10, 10))
ne_roads.plot(ax=ax1)
ne_roads_clip.plot(ax=ax2)
ax1.set_title("Unclipped roads")
ax2.set_title("Clipped roads")
ax1.set_axis_off()
ax2.set_axis_off()
plt.axis('equal')
plt.show()
Plot the cropped data.
# Plot the data
fig, ax = plt.subplots(figsize=(12, 8))
country_boundary_us.plot(alpha=1,
color="white",
edgecolor="black",
ax=ax)
ne_roads_clip.plot(ax=ax)
ax.set_axis_off()
plt.axis('equal')
plt.show()
How Clip() works:
Here are the steps involved with clipping data in geopandas - these steps are completed when you use the clip()
function from GeoPandas. This is an oversimplification! If you want to see the actual code, you can look at the code here to see what is happening under the hood. If you look at the code, you’ll notice that points, line, and polygons all have to be handled differently for the clip function to work, which is part of the reason that clip()
is so convenient!
- Subset the roads data using a spatial index.
- Clip the geometry using
.intersection()
- Remove all rows in the geodataframe that have no geometry (this is explained below).
- Update the original roads layer to contained only the clipped geometry
# Create a single polygon object for clipping
poly = clip_obj.geometry.unary_union
spatial_index = shp.sindex
# Create a box for the initial intersection
bbox = poly.bounds
# Get a list of id's for each road line that overlaps the bounding box and subset the data to just those lines
sidx = list(spatial_index.intersection(bbox))
shp_sub = shp.iloc[sidx]
# Clip the data - with these data
clipped = shp_sub.copy()
clipped['geometry'] = shp_sub.intersection(poly)
# Return the clipped layer with no null geometry values
final_clipped = clipped[clipped.geometry.notnull()]
What Does Simplify Do?
fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(10, 10))
# Set a larger tolerance yields a blockier polygon
country_boundary_us.simplify(2, preserve_topology=True).plot(ax=ax1)
# Set a larger tolerance yields a blockier polygon
country_boundary_us.simplify(.2, preserve_topology=True).plot(ax=ax2)
ax1.set_title(
"Data with a higher tolerance value will become visually blockier as there are fewer vertices")
ax2.set_title(
"Data with a very low tolerance value will look smoother but will take longer to process")
ax1.set_axis_off()
ax2.set_axis_off()
plt.show()
# Plot the data by attribute
fig, ax = plt.subplots(figsize=(12, 8))
country_boundary_us.plot(alpha=1,
color="white",
edgecolor="black",
ax=ax)
ne_roads_clip.plot(ax=ax,
column='type',
legend=True)
ax.set_axis_off()
plt.axis('equal')
plt.show()
Below, you create a custom legend. There are many different approaches to this. One is below.
To begin you create a python dictionary for each attribute value in your legend. Below you will see each road type has a dictionary entry and two associated values - a color and a number representing the width of the line in your legend.
'Beltway': ['black', 2]
Color the line for beltway black with a line width of 2.
Next you loop through the dictionary to plot the data. In the loop below, you select each attribute value, and plot it using the color and line width assigned in the dictionary above.
for ctype, data in ne_roads_clip.groupby('type'):
data.plot(color=road_attrs[ctype][0],
label=ctype,
linewidth=road_attrs[ctype][1],
ax=ax)
Finally, a call to ax.legend()
renders the legend using the colors applied in the loop above.
# Plot with a custom legend
# First, create a dictionary with the attributes of each legend item
road_attrs = {'Beltway': ['black', 2],
'Secondary Highway': ['grey', .5],
'Road': ['grey', .5],
'Bypass': ['grey', .5],
'Ferry Route': ['grey', .5],
'Major Highway': ['black', 1]}
# Plot the data
fig, ax = plt.subplots(figsize=(12, 8))
for ctype, data in ne_roads_clip.groupby('type'):
data.plot(color=road_attrs[ctype][0],
label=ctype,
ax=ax,
linewidth=road_attrs[ctype][1])
country_boundary_us.plot(alpha=1, color="white", edgecolor="black", ax=ax)
ax.legend(frameon=False,
loc = (0.1, -0.1))
ax.set_title("United States Roads by Type", fontsize=25)
ax.set_axis_off()
plt.axis('equal')
plt.show()
Leave a Comment