Hea*_*ker 5 python shapely geopandas
我有一个包含一列 shapely.polygons 的 GeoDataFrame。其中一些是不同的,一些则不是:
In [1]: gdf
Out[2]:
geometry
1 POLYGON ((1 1, 1 2, 2 2, 2 1, 1 1))
2 POLYGON ((1 3, 1 4, 2 4, 2 3, 1 3))
3 POLYGON ((1 1, 1 2, 2 2, 2 1, 1 1))
4 POLYGON ((3 1, 3 2, 4 2, 4 1, 3 1))
5 POLYGON ((1 3, 1 4, 2 4, 2 3, 1 3))
Run Code Online (Sandbox Code Playgroud)
我只需要找到不同的(不重叠的)多边形:
In [1]: gdf_distinct
Out[2]:
geometry
1 POLYGON ((1 1, 1 2, 2 2, 2 1, 1 1))
2 POLYGON ((1 3, 1 4, 2 4, 2 3, 1 3))
4 POLYGON ((3 1, 3 2, 4 2, 4 1, 3 1))
Run Code Online (Sandbox Code Playgroud)
由于多边形不可散列,我无法在 Pandas 中使用简单的方法:
In [1]: gdf_distinct = gdf['geometry'].unique()
TypeError: unhashable type: 'Polygon'
Run Code Online (Sandbox Code Playgroud)
是否有任何简单有效的方法来获得仅包含不同多边形的新 GeoDataFrame?
附:
我找到了一种方法,但它仅适用于完全重复的多边形,而且我认为效率不是很高:
In [1]: m = []
for index, row in gdf.iterrows():]
if row['geometry'] not in m:
m.append(row['geometry'])
gdf_distinct = GeoDataFrame(geometry=m)
Run Code Online (Sandbox Code Playgroud)
让我们从 4 个多边形的列表开始,其中三个与其他多边形重叠:
from shapely.geometry import Polygon
import geopandas
polygons = [
Polygon([[1, 1], [1, 3], [3, 3], [3, 1], [1, 1]]),
Polygon([[1, 3], [1, 5], [3, 5], [3, 3], [1, 3]]),
Polygon([[2, 2], [2, 3.5], [3.5, 3.5], [3.5, 2], [2, 2]]),
Polygon([[3, 1], [3, 2], [4, 2], [4, 1], [3, 1]]),
]
gdf = geopandas.GeoDataFrame(data={'A': list('ABCD')}, geometry=polygons)
gdf.plot(column='A', alpha=0.75)
Run Code Online (Sandbox Code Playgroud)
它们看起来像这样:
因此,我们可以循环遍历每一个,然后循环遍历所有其他并检查与shapelyAPI 的重叠。如果没有任何重叠,我们会将其附加到我们的输出列表中:
non_overlapping = []
for p in polygons:
overlaps = []
for g in filter(lambda g: not g.equals(p), polygons):
overlaps.append(g.overlaps(p))
if not any(overlaps):
non_overlapping.append(p)
Run Code Online (Sandbox Code Playgroud)
任何给我的:
['POLYGON ((3 1, 3 2, 4 2, 4 1, 3 1))']
Run Code Online (Sandbox Code Playgroud)
这正是我所期望的。
但这实际上是O(N^2),而且我认为不必如此。
所以我们尽量不要两次检查同一对:
non_overlapping = []
for n, p in enumerate(polygons[:-1], 1): # don't include the last element
overlaps = []
for g in polygons[n:]: # loop from the next element to the end
overlaps.append(g.overlaps(p))
if not any(overlaps):
non_overlapping.append(str(p))
Run Code Online (Sandbox Code Playgroud)
我得到了相同的结果,并且在我的机器上速度更快了一点。
if我们可以通过在语句中使用生成器而不是普通块来稍微压缩循环for:
non_overlapping = []
for n, p in enumerate(polygons[:-1], 1):
if not any(p.overlaps(g) for g in polygons[n:]):
non_overlapping.append(p)
Run Code Online (Sandbox Code Playgroud)
相同的故事。
| 归档时间: |
|
| 查看次数: |
2612 次 |
| 最近记录: |