from shapely.geometry import Point, LineString
from shapely.ops import snap, split, nearest_points
def clipLine(ls, p1, p2):
p1 = nearest_points(ls, p1)[0]
geom = snap(ls, p1, 1)
parts = split(geom, p1).geoms # 新版本的shapely对该函数的返回值做了改动
if len(parts) == 1:
fp = parts[0]
else:
fp = parts[0] if parts[0].distance(p2) < parts[1].distance(p2) else parts[1]
p2 = nearest_points(fp, p2)[0]
geom = snap(fp, p2, 0.1)
parts = split(geom, p2).geoms #
if len(parts) == 1:
res = parts[0]
else:
res = parts[0] if parts[0].distance(p1) < parts[1].distance(p1) else parts[1]
return res
if __name__ == '__main__':
ls = LineString([(0, 0), (5, 5)])
p1 = Point((1, 1))
p2 = Point([3, 3])
res = clipLine(ls, p1, p2)
print(res.wkt) # LINESTRING (1 1, 3 3)
