以下将找到 中的(x1, y1)
每个点之间的(欧几里得)距离p
:
In [6]: [math.sqrt((x1-x2)**2+(y1-y2)**2) for x2,y2 in p]
Out[6]: [0.0, 1.0, 1.0, 1.4142135623730951, 1.0, 1.4142135623730951]
如果您只想要距 一定距离内的点(x1, y1)
,您可以编写:
In [8]: [(x2,y2) for x2,y2 in p if math.sqrt((x1-x2)**2+(y1-y2)**2) <= 1.0]
Out[8]: [(0, 0), (1, 0), (0, 1), (0, -1)]
这里,1.0
是所需的半径。
把它们放在一起:
import math
def filter_points(points, origin, radius=1.0):
x1, y1 = origin
return [(x2,y2) for x2,y2 in points if math.sqrt((x1-x2)**2+(y1-y2)**2) <= radius]
p = [(0, 0), (1, 0), (0, 1), (1, 1), (0, -1), (-1, 1)]
print(filter_points(p, (0, 0), 1.0))
NB 值得牢记四舍五入问题:非常接近边界的点最终可能会被错误分类。它是否重要,以及如何最好地处理它取决于您打算如何处理结果。