无循环计算单点与LineString列表间的最近点

问题背景 #

给定单个Point对象和一组LineString对象，需要计算该点与每个LineString的最近点对。直接用shapely.ops.nearest_points做纯Python循环遍历，在处理大量线对象时效率极低，因此需要更高效的实现方式。

示例数据 #

from shapely.geometry import LineString, Point

line1 = LineString([(0, 0), (1, 1)])
line2 = LineString([(0, 1), (1, 0)])
line3 = LineString([(1, 0), (1, 1)])

line_list = [line1, line2, line3]
point = Point([2, 2])

高效实现方案 #

利用geopandas的向量化处理能力（底层基于numpy和C扩展，远快于纯Python循环），实现批量计算：

import geopandas as gpd
from shapely.ops import nearest_points

def get_nearest_points(point, line_list):
    # 将线列表转换为GeoSeries，以便进行向量化操作
    lines_geo_series = gpd.GeoSeries(line_list)
    # 批量计算每个线与目标点的最近点对
    nearest_pairs = lines_geo_series.apply(lambda line: nearest_points(point, line))
    # 转换为列表格式返回
    return nearest_pairs.tolist()

# 调用函数并输出结果
result = get_nearest_points(point, line_list)
print(result)

输出结果 #

[(<POINT (2 2)>, <POINT (1 1)>), (<POINT (2 2)>, <POINT (0.5 0.5)>), (<POINT (2 2)>, <POINT (1 1)>)]

性能说明 #

相比纯Python循环，geopandas的apply方法通过底层优化大幅提升了处理效率，当你需要处理成百上千个LineString时，这个方案的优势会非常明显。

内容的提问来源于stack exchange，提问作者ulicious