我的代码可以生成点之间所有可能的行:
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()
但我希望每条线相交多少个点。 (分为蓝色)
答案 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,则表示您有一个交点。