首页 > 代码库 > 均匀的生成圆和三角形内的随机点
均匀的生成圆和三角形内的随机点
代码在每一章节最后
一、均匀生成圆内的随机点
我们知道生成矩形内的随机点比较容易,只要分别随机生成相应的横坐标和纵坐标,比如随机生成范围[-10,10]内横坐标x,随机生成范围[-20,20]内的纵坐标y,那么(x,y)就是生成的随机点。由此,我们很容易的想到了算法1
算法1(正确的):
每个圆对应一个外切矩形,我们随机生成矩形内的点,如果该点在圆内,就返回改点,否则重新生成直到生成的点在圆内。
该方法的缺点是有可能连续几次都生成不了符合要求的点。(可以求得:每次生成点时,该点有 的概率在圆内)
算法2(错误的):
可能有的人想到该方法,根据圆的解析式 (假设圆心在原点)我们可以先随机生成[-R, R]范围内横坐标x,然后生成 范围内的随机数y,(x,y)就是需要的点。
我们写程序模拟了该过程,从下图可以看出,我们可以看到当x靠近圆的边缘使,y的范围减小,因此两边边缘的点较密集,靠近圆心的点较稀疏。
算法3(错误的):
还可以根据极坐标,圆内的点可以如下描述
x = r*sin(theta)
y = r*cos(theta)
其中0 <= r <= R, 0 <= theta < 360
先随机生成[0, 360)内的theta,然后随机生成[0, R]内的r。
theta固定后,当r越靠近R,即点越靠近圆的边缘,因此如果r是[0,R]等概率生成的,那么圆的边缘的点会比靠近圆心处要稀疏。
算法4(正确的):
和算法3一样还是根据极坐标
x = r*sin(theta)
y = r*cos(theta)
其中0 <= r <= R, 0 <= theta < 360 本文地址
先随机生成[0, 360)内的theta,然后随机生成[0, R]内的k, 且r = sqrt(k)。
这样使得r的分布在靠近边缘的地方点变得较密,具体证明可参考here, 也可以参考论文“矩形和椭圆内均匀分布随机点定理及应用”,圆是椭圆的特例
以上的4个算法的代码如下(Python3.3):
import numpy as npimport matplotlib.pyplot as pltimport randomimport math#博客算法1def GeneratePointInCycle1(point_num, radius): for i in range(1, point_num+1): while True: x = random.uniform(-radius, radius) y = random.uniform(-radius, radius) if (x**2) + (y**2) < (radius**2): break plt.plot(x, y, ‘*‘, color = "black")#博客算法2def GeneratePointInCycle2(point_num, radius): for i in range(1, point_num+1): x = random.uniform(-radius, radius) y_max = math.sqrt(radius**2 - x**2) y = random.uniform(-y_max, y_max) plt.plot(x, y, ‘*‘, color = "black")#博客算法3def GeneratePointInCycle3(point_num, radius): for i in range(1, point_num+1): theta = random.random()*2*pi; r = random.uniform(0, radius) x = r*math.sin(theta) y = r*math.cos(theta) plt.plot(x, y, ‘*‘, color = "black")#博客算法4def GeneratePointInCycle4(point_num, radius): for i in range(1, point_num+1): theta = random.random()*2*pi; r = random.uniform(0, radius) x = math.sin(theta)* (r**0.5) y = math.cos(theta)* (r**0.5) plt.plot(x, y, ‘*‘, color = "black") pi = np.pitheta = np.linspace(0, pi*2, 1000)R = 1x = np.sin(theta)*Ry = np.cos(theta)*Rplt.figure(figsize=(6,6))plt.plot(x,y,label = "cycle",color="red",linewidth=2)plt.title("cycyle")GeneratePointInCycle4(4000, R) #修改此处来显示不同算法的效果plt.legend()plt.show()
一、均匀生成三角形内的随机点
算法5(错误的)
对于三角形ABC和一点P,可以有如下的向量表示:
如果p点在三角形内部,则满足:1 >= u >= 0, 1 >= v >= 0, u+v <= 1。
先生成[0,1]的随机数u,然后生成[0, 1-u]内的随机数v,u、v生成后,就可以得到p点的坐标:
由下图可知,该算法生成的点在靠近A点处较浓密
算法6(正确的)
如图所示,三角形ABC有与之对应的矩形ABNM,且矩形面积是三角形的两倍,三角形ADC和CMA全等,CDB和BNC全等。
我们可以先生成矩形ABNM内的随机点P,如果P刚好在三角形ABC中,那么符合要求;如果P不在三角形ABC中,P要么在AMC中,要么在BNC中,如图P在BNC中,我们求P关于BC中点的的中心对称点,该点一定在三角形中。P在AMC中同理。这样可以保重三角形外的点都可以均匀的一一对应到三角形内部。
后面的代码中,为了简化计算,我们假设AB是平行X轴的。
对于生成任意多边形内的随机点,我们可以把它分割成三角形,再来生成随机点。
算法5和算法6的代码如下(Python3.3):
import numpy as npimport matplotlib.pyplot as pltimport matplotlib.lines as lineimport randomimport math#对应博客算法5def GeneratePointInTriangle1(point_num, pointA, pointB, pointC): for i in range(1, point_num+1): u = random.uniform(0.0, 1.0) v = random.uniform(0.0, 1.0 - u) pointP = u*pointC + v*pointB + (1.0-u-v)*pointA; plt.plot(pointP[0], pointP[1], ‘*‘, color = "black")#根据向量叉乘计算三角形面积,参考 http://www.cnblogs.com/TenosDoIt/p/4024413.htmldef ComputeTriangleArea(pointA, pointB, pointC): return math.fabs(np.cross(pointB - pointA, pointB - pointC)) / 2.0#判断点P是否在三角形ABC内,参考 http://www.cnblogs.com/TenosDoIt/p/4024413.htmldef IsPointInTriangle(pointA, pointB, pointC, pointP): area_abc = ComputeTriangleArea(pointA, pointB, pointC) area_pab = ComputeTriangleArea(pointA, pointB, pointP) area_pbc = ComputeTriangleArea(pointP, pointB, pointC) area_pac = ComputeTriangleArea(pointP, pointA, pointC) return math.fabs(area_pab + area_pac + area_pbc - area_abc) < 0.000001#计算一个点关于某一点的中心对称点def ComputeCentralSymmetryPoint(point_src, point_center): return np.array([point_center[0]*2-point_src[0], point_center[1]*2-point_src[1]])#对应博客算法6def GeneratePointInTriangle2(point_num, pointA, pointB, pointC): for i in range(1, point_num+1): pointP = np.array([random.uniform(pointA[0], pointB[0]), random.uniform(pointA[1], pointC[1])]) if not IsPointInTriangle(pointA, pointB, pointC, pointP): if pointP[0] > pointC[0]: pointP = ComputeCentralSymmetryPoint(pointP, np.array([(pointC[0] + pointB[0])/2, (pointC[1] + pointB[1])/2])) else: pointP = ComputeCentralSymmetryPoint(pointP, np.array([(pointC[0] + pointA[0])/2, (pointC[1] + pointA[1])/2])) plt.plot(pointP[0], pointP[1], ‘*‘, color = "black") fig = plt.figure()#三角形三个顶点pointA = np.array([0,1])pointB = np.array([3,1])pointC = np.array([1,2])plt.plot([pointA[0],pointB[0]], [pointA[1],pointB[1]])plt.plot([pointA[0],pointC[0]], [pointA[1],pointC[1]])plt.plot([pointB[0],pointC[0]], [pointB[1],pointC[1]])GeneratePointInTriangle2(1500, pointA, pointB, pointC) #修改此处来显示不同算法的效果plt.ylim(0.5,2)plt.xlim(0,3)plt.show()
【版权声明】转载请注明出处:http://www.cnblogs.com/TenosDoIt/p/4025221.html
均匀的生成圆和三角形内的随机点