[英]How to broadcast this euclidean distance?
所以,我有这段代码,我试图计算 list1 上每个元素的欧氏距离,如果 list1 有 2 个元素,它会抛出错误,对此有什么想法吗?
import numpy as np
from scipy.spatial import distance
list1 =[(10.2,20.2),(5.3,9.2)]
list2 = [(2.2,3.3)]
list1 =np.array(list1)
dist1= distance.euclidean(list1,list2)
print("distance",dist1)
印刷:
ValueError: Input vector should be 1-D.
您可以直接操作numpy
arrays 以在此处找到欧氏距离。
我假设list1
或list2
包含1
元素,并且要计算另一个列表的每个元素与单个元素之间的距离。 Rest 由numpy 广播处理。
import numpy as np
list1 =[(10.2,20.2),(5.3,9.2)]
list2 = [(2.2,3.3)]
a = np.array(list1)
b = np.array(list2)
dist = np.sqrt(((b - a)**2).sum(axis = 1))
dist
:距离
array([18.69786084, 6.66483308])
其中dist[0]
给出distance(list1[0], list2[0])
并且dist[1]
给出distance(list1[1], list2[0])
。
即使list1
具有任意数量的点,它也可以概括,唯一的限制是另一个列表应该只有一个点。
正如您所说,for 循环在这里可能是最简单的。 我已将您的列表更改为列表列表而不是元组列表,尽管我不确定这是否真的有必要。 我没有安装 scipy 来检查。
from scipy.spatial import distance
list1 =[[10.2,20.2],[5.3,9.2]]
list2 = [2.2,3.3]
for point in list1:
dist1= distance.euclidean(point,list2)
print("distance",dist1)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.