PyGEOS ExtensionArray for pandas
Project description
Small wrapper to use pygeos functions from pandas.
The main use case is if you want to have geometries in your dataframe,
but you do not care about CRS at all and thus do not need all the extra features from GeoPandas.
Example
Let's get started by first creating a dataframe with PyGEOS data.
Note that we need to explicitly set the type of the PyGEOS columns to "geos"!
>>> import pandas as pd
>>> import pygeos
>>> import pgpd
>>> # Create a DataFrame
>>> df = pd.DataFrame({
... 'a': list('abcde'),
... 'poly': pygeos.box(range(5), 0, range(10,15), 10),
... 'pt': pygeos.points(range(5), range(10,15))
... })
>>> df = df.astype({'poly':'geos', 'pt':'geos'})
>>> df
a poly pt
0 a POLYGON ((0 0, 0 10, 10 10, 10 0, 0 0)) POINT (0 10)
1 b POLYGON ((1 0, 1 10, 11 10, 11 0, 1 0)) POINT (1 11)
2 c POLYGON ((2 0, 2 10, 12 10, 12 0, 2 0)) POINT (2 12)
3 d POLYGON ((3 0, 3 10, 13 10, 13 0, 3 0)) POINT (3 13)
4 e POLYGON ((4 0, 4 10, 14 10, 14 0, 4 0)) POINT (4 14)
We can access pygeos functionality through the "geos" accessor namespace.
PyGEOS functions that don't require any arguments besides the geometries are accessed as properties on the Series,
others are accessed as methods.
>>> df.poly.geos.length
0 40.0
1 40.0
2 40.0
3 40.0
4 40.0
Name: length, dtype: float64
>>> df.pt.geos.total_bounds
xmin 0.0
ymin 10.0
xmax 4.0
ymax 14.0
Name: total_bounds, dtype: float64
>>> df.poly.geos.clip_by_rect(0, 0, 5, 10)
0 POLYGON ((0 0, 0 10, 5 10, 5 0, 0 0))
1 POLYGON ((1 0, 1 10, 5 10, 5 0, 1 0))
2 POLYGON ((2 0, 2 10, 5 10, 5 0, 2 0))
3 POLYGON ((3 0, 3 10, 5 10, 5 0, 3 0))
4 POLYGON ((4 0, 4 10, 5 10, 5 0, 4 0))
Name: clip_by_rect, dtype: geos
While all PyGEOS functions are available on Series, some are made available on the DataFrame as well.
The functions that are available on DataFrames are those that have a 1-to-1 mapping (create one output for each geometry in the column), or those that have a fixed number of outputs for the entire geos column.
>>> # Fixed number of outputs (ic. xmin,ymin,xmax,ymax)
>>> df.geos.total_bounds
poly pt
xmin 0.0 0.0
ymin 0.0 10.0
xmax 14.0 4.0
ymax 10.0 14.0
>>> # For every PyGEOS function that has a 1-to-1 relation,
>>> # the DataFrame variant allows inplace modification
>>> df.geos.apply(lambda coord: coord*2, inplace=True)
>>> df
a poly pt
0 a POLYGON ((0 0, 0 20, 20 20, 20 0, 0 0)) POINT (0 20)
1 b POLYGON ((2 0, 2 20, 22 20, 22 0, 2 0)) POINT (2 22)
2 c POLYGON ((4 0, 4 20, 24 20, 24 0, 4 0)) POINT (4 24)
3 d POLYGON ((6 0, 6 20, 26 20, 26 0, 6 0)) POINT (6 26)
4 e POLYGON ((8 0, 8 20, 28 20, 28 0, 8 0)) POINT (8 28)
Finally, PyGEOS also has some binary functions, which work on 2 different sets of geometries.
These functions are also made available on Series, but work slightly differently.
We added a manner
argument, which can be one of 3 different values: keep, align, expand.
This argument dictates how the 2 sets of geometries are transformed before running the binary function:
- keep: Function is run on the input as is.
- align: Align both sets with each other, according to their index (only works when
other
is a Series). - expand: Expand both sets to a 2D array and compare each geometry of set A with each geometry of set B (returns a 2D array of dimension <len(A), len(B)>).
>>> # KEEP: Just runs the `contains` function on the "poly" column data and the given Point
>>> df.poly.geos.contains(pygeos.Geometry("Point (23 10)"), manner='keep')
0 False
1 False
2 True
3 True
4 True
Name: contains, dtype: bool
>>> # ALIGN: We only pass 3 points, but tell the function to align the data according to the index
>>> df.poly.geos.distance(df.pt[1:4], manner='align')
0 NaN
1 2.0
2 4.0
3 6.0
4 NaN
Name: distance, dtype: float64
>>> # EXPAND: Compare each polygon with each point (returns numpy.ndarray <5,3> in this case)
>>> df.poly.geos.distance(df.pt[1:4], manner='expand')
array([[2. , 4. , 6. ],
[2. , 4. , 6. ],
[2.82842712, 4. , 6. ],
[4.47213595, 4.47213595, 6. ],
[6.32455532, 5.65685425, 6.32455532]])
One last difference is that you can omit the other
set of geometries.
The method will then automatically choose the expand mode and use the self
data twice.
>>> # Compute all possible intersection areas of the geometries in the "poly" column
>>> pygeos.area(df.poly.geos.intersection())
array([[400., 360., 320., 280., 240.],
[360., 400., 360., 320., 280.],
[320., 360., 400., 360., 320.],
[280., 320., 360., 400., 360.],
[240., 280., 320., 360., 400.]])
GeoPandas
The main use case for this library is not to depend on GeoPandas and all of its dependencies. However -if you need to- this library provides methods to convert from and to GeoPandas.
DataFrame
>>> gdf = df.geos.to_geopandas(geometry='poly', crs='WGS84')
>>> gdf
a poly pt
0 a POLYGON ((0.00000 0.00000, 0.00000 20.00000, 2... POINT (0 20)
1 b POLYGON ((2.00000 0.00000, 2.00000 20.00000, 2... POINT (2 22)
2 c POLYGON ((4.00000 0.00000, 4.00000 20.00000, 2... POINT (4 24)
3 d POLYGON ((6.00000 0.00000, 6.00000 20.00000, 2... POINT (6 26)
4 e POLYGON ((8.00000 0.00000, 8.00000 20.00000, 2... POINT (8 28)
>>> gdf.dtypes
a object
poly geometry
pt geos
dtype: object
>>> df2 = gdf.geos.from_geopandas()
>>> df2
a poly pt
0 a POLYGON ((0 0, 0 20, 20 20, 20 0, 0 0)) POINT (0 20)
1 b POLYGON ((2 0, 2 20, 22 20, 22 0, 2 0)) POINT (2 22)
2 c POLYGON ((4 0, 4 20, 24 20, 24 0, 4 0)) POINT (4 24)
3 d POLYGON ((6 0, 6 20, 26 20, 26 0, 6 0)) POINT (6 26)
4 e POLYGON ((8 0, 8 20, 28 20, 28 0, 8 0)) POINT (8 28)
>>> df2.dtypes
a object
poly geos
pt geos
dtype: object
Series
>>> gs = df.pt.geos.to_geopandas(crs='WGS84')
>>> gs
0 POINT (0.00000 20.00000)
1 POINT (2.00000 22.00000)
2 POINT (4.00000 24.00000)
3 POINT (6.00000 26.00000)
4 POINT (8.00000 28.00000)
Name: pt, dtype: geometry
>>> s2 = gs.geos.from_geopandas()
>>> s2
0 POINT (0 20)
1 POINT (2 22)
2 POINT (4 24)
3 POINT (6 26)
4 POINT (8 28)
Name: pt, dtype: geos
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.