geopandas中的分组形式

Question

我有一个geopandas数据框，其中的geom字段保留了异构的形状数据类型。我想按类型（例如shapely.geometry.Point，shapely.geometry.Polygon，shapely.geometry.MultiPolygon等）对它们进行分组。此任务将通过以下方式完成：

main_gdf_clean['geometry'].groupby(by=shapely.geometry.Point)

是否有可行的方法来完成我正在描述的内容？

Answer 1

GeoSeries.geom_type

按几何类型（df['geometry'].geom_type）而不是实际几何进行分组。

import geopandas as gpd
from shapely.geometry import Polygon, Point

df  = gpd.GeoSeries([Polygon([(1,1), (3,1), (3,3), (1,3)]),
                     Polygon([(3,3), (5,3), (5,5), (3,5)]),
                     Point(1,2), Point(2,3), Point(3,3)]).to_frame('geometry')
df = df.set_geometry(col='geometry')  # So this example knows geometry

# Broadcast # of unique shapes
df['N_items'] = df.groupby(df['geometry'].geom_type).transform('count')
#                                            geometry  N_items
#0  POLYGON ((1.00000 1.00000, 3.00000 1.00000, 3....        2
#1  POLYGON ((3.00000 3.00000, 5.00000 3.00000, 5....        2
#2                            POINT (1.00000 2.00000)        3
#3                            POINT (2.00000 3.00000)        3
#4                            POINT (3.00000 3.00000)        3