http://www.infocool.net/kb/Python/201609/191671.html
点与点之间, 线与线之间,点与线之间的位置关系是一类非常重要的问题。它不仅是平面几何学的基石,也常常应用于LBS(Location Based Service),社交网络,以及数据库查询等领域。
本文中,我将给出判断这些关系的相关算法,作为参考。需要说明的是,我给出的这些问题的解法,都是建立在二维平面空间之上。有关多维空间的位置关系,大家可以仿照二维空间中问题的思路,做相应的拓展。
语言上,我用的当然还是Python.
点与点之间的距离
先从最简单的点与点的位置关系说起。一般情况下,我们只关心点与点之间的距离。
1. 点类的定义
为使算法思路更加清晰,先定义点类 Point
,既然是在二维空间上,那么每个点都应该有两个属性:x, y分别代表点的横纵坐标。
class Point(object):
"""Point are two-dimension"""
def __init__(self, x, y):
self.x = x
self.y = y
接下来就看看如何计算两点之间距离:当然可以用初中学的欧氏距离最基本的计算方法。但是考虑到代码编写的效率,以及方便以后向高维空间拓展。我在本文中将尽量使用向量计算。
而为了简化代码,我们使用对于向量运算已经相当成熟的库numpy
2. 两点之间距离的计算
显然,两点可以构成向量,而向量的长度则是其内积的开方。空间中,点 A 与点 B 的距离可以用向量 AB−→ 的模 |AB−→| 表示。所以,现在需要做的,就是写一个函数,以两点为参数,计算由这两点构成的向量的模。
为了和本文之后的问题保持编码风格上一致,同时简化代码编写。我使用对向量运算已经极为成熟的库numpy帮助计算。并且定义了一个新的类 Vector
,类 Vector
以向量的起点和终点作为输入,生成一个只拥有属性x和y的向量对象。
最后,和前面定义的类放在一起,代码如下:
import numpy as np
class Point(object):
"""Point are two-dimension"""
def __init__(self, x, y):
self.x = x
self.y = y
class Vector(object):
"""start and end are two points"""
def __init__(self, start, end):
self.x = end.x - start.x
self.y = end.y - start.y
def pointDistance(p1, p2):
"""calculate the distance between point p1 and p2"""
v = Vector(p1, p2)
t = np.array([v.x, v.y])
return float(np.sqrt(t @ t))
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
说明一下,在Python3.5以后的版本中,使用numpy库时,ndarray对象之间的乘法可以用 @
,代替之前的 v1.dot(v2)
这样的形式。
点与线之间的位置关系
1. 线的分类
点与线之间的位置关系就要稍微复杂一些了,复杂之处在于线分线段和直线两种情况。但是,在定义类的时候我都用两点来代表线段(直线)的两个属性。于是,至少代码看上去是没什么分别的。
不同之处在于,线段的两个点事两个端点,而直线的两个点是直线上任意两点。
class Segment(object):
"""the 2 points p1 and p2 are unordered"""
def __init__(self, p1, p2):
self.p1 = p1
self.p2 = p2
class Line(object):
"""p1 and p2 are 2 points in straight line"""
def __init__(self, p1, p2):
self.p1 = p1
self.p2 = p2
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
需要注意的是,这里并没有说线段的两个点是什么顺序(不一定说左边的点就是p1,右边就是p2)
2. 点与线的位置关系
(1) 计算点到直线的距离
如Fig.1(a)所示,现要求点 C 到直到直线 AB 的距离。还是向量法,据向量知识可知:
cos∠CAB=AC−→⋅AB−→|AC−→|⋅|AB−→|
再由三角形知识可知,线段 AD 的长度为:
|AC−→|⋅cos∠CAB
所以, AD−→− 可以这样计算:
AD−→−=AB−→|AB−→|⋅|AD−→−|=AB−→|AB−→|⋅|AC−→|⋅cos∠CAB=AB−→⋅AC−→|AB−→|2AB−→
当 AD−→− 计算完成之后,可以根据 AD−→− 相应的坐标值得到点 D 的坐标,再由上面点和点之间的距离,即可得到线段 CD 的长度。
给出完整的代码如下:
import numpy as np
class Point(object):
"""Point are two-dimension"""
def __init__(self, x, y):
self.x = x
self.y = y
class Segment(object):
"""the 2 points p1 and p2 are unordered"""
def __init__(self, p1, p2):
self.p1 = p1
self.p2 = p2
class Line(object):
"""p1 and p2 are 2 points in straight line"""
def __init__(self, p1, p2):
self.p1 = p1
self.p2 = p2
class Vector(object):
"""start and end are two points"""
def __init__(self, start, end):
self.x = end.x - start.x
self.y = end.y - start.y
def pointDistance(p1, p2):
"""calculate the distance between point p1 and p2"""
v = Vector(p1, p2)
t = np.array([v.x, v.y])
return float(np.sqrt(t @ t))
def pointToLine(C, AB):
"""calculate the shortest distance between point C and straight line AB, return: a float value"""
vector_AB = Vector(AB.p1, AB.p2)
vector_AC = Vector(AB.p1, C)
tAB = np.array([vector_AB.x, vector_AB.y])
tAC = np.array([vector_AC.x, vector_AC.y])
tAD = ((tAB @ tAC) / (tAB @ tAB)) * tAB
Dx, Dy = tAD[0] + AB.p1.x, tAD[1] + AB.p1.y
D = Point(Dx, Dy)
return pointDistance(D, C)
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
- 66
- 67
- 68
(2) 判断点是否在直线上
既然已经能够计算点到直线的距离了,那么,只需要看点到直线的距离是否为0即可知道这个点在不在直线上。
接着上面的代码,可以写出如下函数:
def pointInLine(C, AB):
"""determine whether a point is in a straight line"""
return pointToLine(C, AB) < 1e-9
(3) 判断点是否在线段上
处理完了点和直线的位置关系,我们接着来看点与线段的位置关系。其实,最常用的就是一点:判断点是否在线段上。这和判断点是否在直线上最大的区别在于线段有起点、终点。
如Fig.1(b)所示,判断点 C 在不在线段 AB 上,可以这样解决:
-
计算点 C 到线段 AB 所在直线的距离
-
若这个距离为0,继续第3步;否则,返回
False
-
若点 C 的横坐标在点 A 与点 B 的横坐标之间,则返回
True
,否则返回False
函数如下:
def pointInSegment(C, AB):
"""determine whether a point is in a segment"""
if pointInLine(C, Line(AB.p1, AB.p2)):
return min(AB.p1.x, AB.p2.x) <= C.x <= max(AB.p1.x, AB.p2.x)
return False
还是需要结合上面的代码的,这里省略,只是写出函数
线与线之间的位置关系
1. 直线与直线
主要就是判断两条之间是否相交。很容易解决:先把两条直线做成向量,再判断两个向量是否平行即可。
如Fig.1(c)所示,判断直线 AB 与 CD 是否平行,可以通过向量平行的判别公式:
B.y−A.yB.x−A.x=D.y−C.yD.x−C.x
也就是判断两个向量的斜率是否相等。若相等,则平行;若不等,则不平行。
def linesAreParallel(l1, l2):
"""determine whether 2 straight lines l1, l2 are parallel"""
v1 = Vector(l1.p1, l1.p2)
v2 = Vector(l2.p1, l2.p2)
return abs((v1.y / v1.x) - (v2.y / v2.x)) < 1e-9
说明两点:
-
这个函数既能判断直线是否平行,也能判断线段是否平行
-
考虑到斜率在大多情况下是浮点数,所以不能用操作符
==
判断两个浮点数是否相等,而是通过 abs(k1
- k2) < 1e-9
判断。
2. 线段与线段
线段与线段是否平行通过上面的函数可以实现,那现在主要要解决的问题是如何判断两直线是否相交。这也是一个非常经典的算法。
一种比较标准的做法是通过“跨立实验”。基本逻辑是这样,倘若两条线段满足以下两个条件之一,则这两条直线一定相交:
1. 相互跨越对方线段所在的直线
2. 一条线段的端点在另一条直线上
上面的这两个条件包含了所有可能的线段相交的情况,如Fig.2所示。
基于此,可以先想想如何判断一条线段是“跨越”对方所在直线的。不过在此之前,先补充一个知识:向量叉积。
向量叉积是用来判断两个向量之间方向的
他的计算方法如下:
假设两个向量分别为 a⃗ =(x1,y1),b⃗ =(x2,y2) ,则 a⃗ 与 b⃗ 的叉积如下定义:
a⃗ ×b⃗ =x1⋅y2−x2⋅y1
其中,符号 × 用来表示向量叉积运算。而 a⃗ 和 b⃗ 都是以坐标原点为起点,以上面给出的坐标: (x1,y1),(x2,y2) 为终点的两个向量。
根据这个叉积的算法,我们不难得到以下规律:
-
若 a⃗ ×b⃗ >0 ,则 a⃗ 在 b⃗ 的顺时针方向
-
若 a⃗ ×b⃗ <0 ,则 a⃗ 在 b⃗ 的逆时针方向
-
若 a⃗ ×b⃗ =0 ,则 a⃗ 与 b⃗ 共线
其实向量叉积表示的是由这两个向量组成的平行四边形的有向面积。具体的推导我这里省略了。感兴趣的话,可以去学习一下行列式的几何意义。以后如果有机会,我可能也会写成博客。
写出计算叉积的函数:当然要结合上面定义的函数和类,这里只是给出函数
def crossProduct(v1, v2):
"""calculate the cross product of 2 vectors"""
return v1.x * v2.y - v1.y * v2.x
了解了叉积的性质以及实现,我们回过头再看线段的相互跨越问题。如图Fig.2,若线段 Q1Q2 跨越线段 P1P2 所在直线,则无论是Fig.2(a)或Fig.2(b)中的哪种情况,一定有下面的结论成立:
Q1P1−→−−×Q1Q2−→−−∗Q1P2−→−−×Q1Q2−→−−<0
更进一步,如果两条线段相互跨越,如Fig.3(a)所示的那样(Fig.3(b)的情况是线段 P1P2 跨越了线段 Q1Q2 所在的直线,但是反过来,线段 Q1Q2 却没有跨越线段 P1P2 所在的直线)。则相互跨越的两条线段一定同时满足下面的两个条件:
Q1P1−→−−×Q1Q2−→−−∗Q1P2−→−−×Q1Q2−→−−<0
P1Q1−→−−×P1P2−→−−∗P1Q2−→−−×P1P2−→−−<0
此外,别忘了上面红字标出的两线段相交的另一个条件,就是一条线段的一个端点在另一条线段上的情况。比如Fig.3(c),此时,
P2Q1−→−−×P2P1−→−−=0
所以,还需要检测点 Q1 是否在线段 P1P2 上。因为只要 Q1 在 P1P2 所在的直线上,上面的式子就会成立。
综上所述,判断两条线段是否相交的算法思路就很清晰了:
先计算相应向量的叉积,一共是4个结果,根据上面的讲解,分为一下几种情况:
1. 4个结果分为两组;若两组组的结果都是异号的,则一定相交;
2. 4个结果中有0存在,则检查相应的点是否在另一条线段上,在,则相交;不在,则不相交
3. 以上两个条件都不成立,则肯定不相交
最后,我将上面讲的:计算两点的距离;计算点到直线的距离;判断点是否在直线上;判断点是否在线段上;判断两直线(线段)是否平行;判断两条线段是否相交等等计算几何中关于点与线的问题的完整代码写在下面,供大家参考:
点与线之间位置关系的完整代码
import numpy as np
class Point(object):
"""Point are two-dimension"""
def __init__(self, x, y):
self.x = x
self.y = y
class Segment(object):
"""the 2 points p1 and p2 are unordered"""
def __init__(self, p1, p2):
self.p1 = p1
self.p2 = p2
class Line(object):
"""p1 and p2 are 2 points in straight line"""
def __init__(self, p1, p2):
self.p1 = p1
self.p2 = p2
class Vector(object):
"""start and end are two points"""
def __init__(self, start, end):
self.x = end.x - start.x
self.y = end.y - start.y
def pointDistance(p1, p2):
"""calculate the distance between point p1 and p2"""
v = Vector(p1, p2)
t = np.array([v.x, v.y])
return float(np.sqrt(t @ t))
def pointToLine(C, AB):
"""calculate the shortest distance between point C and straight line AB, return: a float value"""
vector_AB = Vector(AB.p1, AB.p2)
vector_AC = Vector(AB.p1, C)
tAB = np.array([vector_AB.x, vector_AB.y])
tAC = np.array([vector_AC.x, vector_AC.y])
tAD = ((tAB @ tAC) / (tAB @ tAB)) * tAB
Dx, Dy = tAD[0] + AB.p1.x, tAD[1] + AB.p1.y
D = Point(Dx, Dy)
return pointDistance(D, C)
def pointInLine(C, AB):
"""determine whether a point is in a straight line"""
return pointToLine(C, AB) < 1e-9
def pointInSegment(C, AB):
"""determine whether a point is in a segment"""
if pointInLine(C, Line(AB.p1, AB.p2)):
return min(AB.p1.x, AB.p2.x) <= C.x <= max(AB.p1.x, AB.p2.x)
return False
def linesAreParallel(l1, l2):
"""determine whether 2 straight lines l1, l2 are parallel"""
v1 = Vector(l1.p1, l1.p2)
v2 = Vector(l2.p1, l2.p2)
return abs((v1.y / v1.x) - (v2.y / v2.x)) < 1e-9
def crossProduct(v1, v2):
"""calculate the cross product of 2 vectors"""
return v1.x * v2.y - v1.y * v2.x
def segmentsIntersect(s1, s2):
"""determine whether 2 segments s1, s2 intersect with each other"""
v1 = Vector(s1.p1, s1.p2)
v2 = Vector(s2.p1, s2.p2)
t1 = Vector(s1.p1, s2.p1)
t2 = Vector(s1.p1, s2.p2)
d1 = crossProduct(t1, v1)
d2 = crossProduct(t2, v1)
t3 = Vector(s2.p1, s1.p1)
t4 = Vector(s2.p1, s1.p2)
d3 = crossProduct(t3, v2)
d4 = crossProduct(t4, v2)
if d1 * d2 < 0 and d3 * d4 < 0:
return True
if d1 == 0:
return pointInSegment(s2.p1, s1)
elif d2 == 0:
return pointInSegment(s2.p2, s1)
elif d3 == 0:
return pointInSegment(s1.p1, s2)
elif d4 == 0:
return pointInSegment(s1.p2, s2)
return False
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
- 66
- 67
- 68
- 69
- 70
- 71
- 72
- 73
- 74
- 75
- 76
- 77
- 78
- 79
- 80
- 81
- 82
- 83
- 84
- 85
- 86
- 87
- 88
- 89
- 90
- 91
- 92
- 93
- 94
- 95
- 96
- 97
- 98
- 99
- 100
- 101
- 102
- 103
- 104
- 105
- 106
- 107
- 108
- 109
- 110
- 111
- 112
- 113
- 114
- 115
- 116
- 117
- 118
- 119
- 120
- 121
- 122
- 123
- 124
- 125
- 126
- 127
- 128
- 129
- 130
- 131
- 132
在上面完成类和函数定义的基础上,给出一个测试脚本,方便检验:
if __name__ == "__main__":
p1 = Point(0, 0)
p2 = Point(2, 2)
print(pointDistance(p1, p2))
l1 = Line(p1, p2)
s1 = Segment(p1, p2)
p3 = Point(3, 3)
print(pointInLine(p3, l1))
print(pointInSegment(p3, s1))
p4 = Point(0, 1)
p5 = Point(2, 3)
l2 = Line(p4, p5)
print(linesAreParallel(l1, l2))
print(pointToLine(p4, l1))
s2 = Segment(Point(0, 2), Point(5, -1))
s3 = Segment(Point(1, 0.7), Point(5, -1))
print(segmentsIntersect(s2, s1))
print(segmentsIntersect(s3, s1))