使用GeoPandas从点创建多边形

Question

我有一个geopandas数据框，其中包含一个形状良好的点几何体列表。还有一列包含ID列表，用于指定每个点所属的唯一多边形。简化输入代码为：

import pandas as pd
from shapely.geometry import Point, LineString, Polygon
from geopandas import GeoDataFrame

data = [[1,10,10],[1,15,20],[1,20,10],[2,30,30],[2,35,40],[2,40,30]] 
df_poly = pd.DataFrame(data, columns = ['poly_ID','lon', 'lat']) 
geometry = [Point(xy) for xy in zip(df_poly.lon, df_poly.lat)]
geodf_poly = GeoDataFrame(df_poly, geometry=geometry)
geodf_poly.head()

我想按多边形ID分组，以便将几何体从点转换为多边形。该输出基本上如下所示：

poly_ID   geometry
1         POLYGON ((10 10, 15 20, 20 10))
2         POLYGON ((30 30, 35 40, 40 30))

我想这很简单，但我很难让它工作。我发现下面的代码允许我将其转换为开放式多段线，但无法计算出多边形。有人能建议如何适应这种情况吗？

geodf_poly = geodf_poly.groupby(['poly_ID'])['geometry'].apply(lambda x: LineString(x.tolist()))

简单地用多边形替换LineString会导致TypeError：类型'Point'的对象没有len（）

Answer 1

在Pandas中完成您的请求有点棘手，因为在您的输出中，您希望文本“POLYGON”，但括号中有数字。

请参见下面的选项

from itertools import chain
df_poly.groupby('poly_ID').agg(list).apply(lambda x: tuple(chain.from_iterable(zip(x['lon'], x['lat']))), axis=1).reset_index(name='geometry')

输出

poly_ID     geometry
0   1   (10, 10, 15, 20, 20, 10)
1   2   (30, 30, 35, 40, 40, 30)

或

from itertools import chain

df_new =df_poly.groupby('poly_ID').agg(list).apply(lambda x: tuple(chain.from_iterable(zip(x['lon'], x['lat']))), axis=1).reset_index(name='geometry')
df_new['geometry']=df_new.apply(lambda x: 'POLYGON ('+str(x['geometry'])+')',axis=1 )
df_new

输出

poly_ID     geometry
0   1   POLYGON ((10, 10, 15, 20, 20, 10))
1   2   POLYGON ((30, 30, 35, 40, 40, 30))

注意：列几何是一个字符串