python:找到两个数组中两点之间的最小距离

Question

我有两个包含一系列元组 (x,y) 的列表，代表笛卡尔平面上的不同点:

a = [(0, 0), (1, 2), (1, 3), (2, 4)]
b = [(3, 4), (4, 1), (5, 3)]

在这种特定情况下，我想找到距离较小的两个点(每个列表一个，不在同一列表内):

[((2, 4), (3, 4))]

其距离等于1。我使用列表理解，如下:

[(Pa, Pb) for Pa in a for Pb in b \
if math.sqrt(math.pow(Pa[0]-Pb[0],2) + math.pow(Pa[1]-Pb[1],2)) <= 2.0]

但这使用了阈值。有没有办法附加 argmin()某处或类似的东西，只得到一对 [((xa, ya), (xb, yb))]最小距离？谢谢。

Answer 1

import numpy
e = [(Pa, Pb) for Pa in a for Pb in b]
e[numpy.argmin([math.sqrt(math.pow(Pa[0]-Pb[0],2) + math.pow(Pa[1]-Pb[1],2)) for (Pa, Pb) in e])]

将按照您的建议使用 argmin 并返回 ((2, 4), (3, 4))