Top
Sp.4ML > Data Engineering  > Toolbox: Get Bounding Box Polygon and centroid of a point cloud in Python
Image with the article title. Decorative purposes only.

Toolbox: Get Bounding Box Polygon and centroid of a point cloud in Python

Let’s assume that we have a set of points, for example:

And we would like to get their Bounding Box and its centroid. How do we achieve it? We can use shapely and GeoPandas packages for it!

from typing import Union

import geopandas as gpd
from shapely.geometry import Polygon


def points_to_bounding_box(points: Union[gpd.GeoSeries, gpd.GeoDataFrame]):
    """
    Function transform a set of points into a Polygon with bounding box.
    
    Parameters
    ----------
    points : gpd.GeoSeries
    
    Returns
    -------
    bounding_box : Polygon
    """
    bounds = points.total_bounds
    
    x_min, y_min, x_max, y_max = bounds
    
    polygon_tuples = [
        (x_min, y_min), (x_max, y_min), (x_max, y_max), (x_min, y_max)
    ]
    
    polygon = Polygon(polygon_tuples)
    return polygon


poly = points_to_bounding_box(centroids)
print(poly)
POLYGON ((1305267.3457557156 210560.55098878054, 2198938.029283345 210560.55098878054, 2198938.029283345 1163583.8925663661, 1305267.3457557156 1163583.8925663661, 1305267.3457557156 210560.55098878054))

If we want to get a centroid, then we can use the centroid attribute of Polygon structure:

print(poly.centroid)
POINT (1752102.6875195305 687072.2217775733)

Plotting all together can be a little bit tricky – we must change simple geometries into GeoSeries objects, and then we can plot our output:

bb = gpd.GeoSeries(poly)
bb.set_crs(crs=centroids.crs)  # Here we must set CRS of our baseline set of points!

cent = bb.centroid  # We can generate centroid from GeoPandas GeoSeries object

# Plot

base = bb.plot(color='white', edgecolor='gray', figsize=(12, 10))
centroids.plot(ax=base, marker='o', color='red', markersize=10);
cent.plot(ax=base, marker='x', color='green', markersize=20)
plt.show()
Point cloud (red dots), bounding box (gray rectangle within the plot), and its centroid (green x) – all generated with GeoPandas and Shapely.
Szymon
Subscribe
Notify of
guest
0 Comments
Oldest
Newest Most Voted
Inline Feedbacks
View all comments
0
Would love your thoughts, please comment.x
()
x