In this recipe, we'll perform a spatial operation to select a subset of a point layer based on the points contained in an overlapping polygon layer. We'll use shapefiles in both cases, with one being a point layer and the other a polygon. This kind of subset is one of the most common GIS operations.
We will need two new shapefiles that have not been used in previous recipes. You can download the point layer from https://geospatialpython.googlecode.com/files/MSCities_Geo_Pts.zip.
Similarly, you can download the geometry layer from https://geospatialpython.googlecode.com/files/GIS_CensusTract.zip.
Unzip these shapefiles and place them in a directory named ms within your qgis_data directory, within your root or home directory.
In this recipe, we will perform several steps to select features in the point layer that fall within the polygon layer, as follows:
- First, load the point layer:
lyrPts = QgsVectorLayer("/qgis_data/ms/MSCities_Geo_Pts.shp", "MSCities_Geo_Pts", "ogr") - Next, load the polygon layer:
lyrPoly = QgsVectorLayer("/qgis_data/ms/GIS_CensusTract_poly.shp", "GIS_CensusTract_poly", "ogr") - Add the layers to the map using a list:
QgsMapLayerRegistry.instance().addMapLayers([lyrPts,lyrPoly]) - Access the polygon layer's features:
ftsPoly = lyrPoly.getFeatures() - Now, iterate through the polygon's features:
for feat in ftsPoly: - Grab each feature's geometry:
geomPoly = feat.geometry() - Access the point features and filter the point features by the polygon's bounding box:
featsPnt = lyrPts.getFeatures(QgsFeatureRequest().setFilterRect(geomPoly.boundingBox())) - Iterate through each point and check whether it's within the polygon itself:
for featPnt in featsPnt: if featPnt.geometry().within(geomPoly):
- If the polygon contains the point, print the point's ID and select the point:
print featPnt.id() lyrPts.select(featPnt.id())
- Now, set the polygon layer as the active map layer:
iface.setActiveLayer(lyrPoly) - Zoom to the polygon layer's maximum extent:
iface.zoomToActiveLayer()
Verify that your map looks similar to the following image:
While QGIS has a number of tools for spatial selection, PyQGIS doesn't have a dedicated API for these types of functions. However, there are just enough methods in the API, thanks to the underlying ogr/GEOS library, that you can easily create your own spatial filters for two layers. Step 7 isn't entirely necessary, but we gain some efficiency using the bounding box of the polygon to limit the number of point features we're examining. Calculations involving rectangles are far quicker than detailed point-in-polygon queries. So, we quickly reduce the number of points we need to iterate through for the more expensive spatial operations.