geoarrow_pyarrow_wkt_to_geopandas_with_area_calculation.py

1import geoarrow.pyarrow as ga
2import geoarrow.pyarrow.compute as gac
3import geopandas as gpd
4
5# Create an array of WKT geometries
6wkt_geometry = [
7    "POLYGON ((0 0, 10 0, 10 10, 0 10, 0 0))",
8    "POLYGON ((10 10, 20 10, 20 20, 10 20, 10 10))"
9]
10array = ga.array(wkt_geometry)
11
12# Compute the area of each geometry using geoarrow compute
13# (Returns a pyarrow.DoubleArray)
14areas = gac.area(array)
15print(f"Areas: {areas.to_pylist()}")
16
17# Convert the GeoArrow array to a GeoPandas GeoDataFrame
18gdf = gpd.GeoDataFrame({"geometry": ga.to_geopandas(array), "area": areas.to_pylist()})
19print(gdf)