如何在两点之间创建线段并找出与线相交的点数?

时间:2011-07-26 19:10:03

标签: python

我的代码可以生成点之间所有可能的行:

import matplotlib.pyplot as plt
import matplotlib.lines as lines
import itertools 

fig=plt.figure()
ax=fig.add_subplot(111)
all_data = [[1,10],[2,10],[3,10],[4,10],[5,10],[3,1],[3,2],[3,3],[3,4],[3,5]]
x, y = zip(*all_data)
plt.scatter(x,y)
for pair in itertools.combinations(all_data,2):
    line=lines.Line2D(*zip(*pair))
    line.set_color('brown')
    ax.add_line(line)

plt.show()

enter image description here

但我希望每条线相交多少个点。 (分为蓝色)

2 个答案:

答案 0 :(得分:1)

也许是这样的(未经测试):

def crossf(p1, p2):
  u"returns a boolean function for testing line (p1, p2)"
  if p1[0] == p2[0]:
    y = [p1[1], p2[1]]
    y.sort()
    # if p3 falls within the y range it is on the line
    return lambda p3: (p3[1] >= y[0]) and (p3[1] <= y[1])
  else:
    slope = p2[1] - p1[1], p2[0] - p1[0]
    # y/x ratio of point (p3 - p1) must equal slope
    return lambda p3: (p3[0] - p1[0]) * slope[1] == (p3[1] - p1[1]) * slope[0]

crosstests = dict([(pair, crossf(*pair) for pair in itertools.combinations(all_data,2)])

for pair in crosstests:
  for point in all_data:
    print 'point %s does %sfall on line %s' % (point, 
        '' if crosstests[pair](point) else 'not ', pair)

答案 1 :(得分:0)

使用此距离指标。 http://mathworld.wolfram.com/Point-LineDistance2-Dimensional.html 它有一个很好的矢量表达到最后。置换所有点线组合,如果d小于epsilon,则表示您有一个交点。