如果您想避免例如PostGIS数据库的重型机械,那么可以使用rtree
包作为(如文档所述)“廉价空间数据库”。这个想法大多如下:
#!/usr/bin/env python
from itertools import product
from random import uniform, sample, seed
from rtree import index
from shapely.geometry import Point, Polygon, box, shape
from shapely.affinity import translate
seed(666)
#generate random polygons, in your case, the polygons are stored
#in geo_data['features']
P = Polygon([(0, 0), (0.5, 0), (0.5, 0.5), (0, 0.5), (0, 0)])
polygons = []
for dx, dy in product(range(0, 100), range(0, 100)):
polygons.append(translate(P, dx, dy))
#construct the spatial index and insert bounding boxes of all polygons
idx = index.Index()
for pid, P in enumerate(polygons):
idx.insert(pid, P.bounds)
delta = 0.5
for i in range(0, 1000):
#generate random points
x, y = uniform(0, 10), uniform(0, 10)
pnt = Point(x, y)
#create a region around the point of interest
bounds = (x-delta, y-delta, x+delta, y+delta)
#also possible, but much slower
#bounds = pnt.buffer(delta).bounds
#the index tells us which polygons are worth checking, i.e.,
#the bounding box of which intersects with the region constructed in previous step
for candidate in idx.intersection(bounds):
P = polygons[candidate]
#test only these candidates
if P.contains(pnt):
print(pnt, P)
只是一个小挑逗,但你实际上并不需要找到的变量。 –