计算多维矩阵中的距离

时间:2017-04-24 15:45:56

标签: python euclidean-distance

我写了这段代码来计算距离。但是得到ValueError: math domain error的错误。

import math

a=[[5, 10],
   [16, 23]]

b=[[6, 1], [21, 2]]

dL=[]
dist=[]
k=0
h=1
for i in range(len(a)):
    dL=[]
    for j in range(len(a)):
        d=math.sqrt((a[i][k]-b[j][k])^2 + (a[i][h]-b[j][h])^2)
        dL.append(d)
    dist.append(dL)

在迭代中,d应为:

i=0, j=0: d=sqrt((5-6)^2+(10-1)^2)
i=0, j=1: d=sqrt((5-21)^2+(10-2)^2)
i=1, j=0: d=sqrt((16-6)^2+(23-1)^2)
i=1, j=1: d=sqrt((16-21)^2+(23-2)^2)

3 个答案:

答案 0 :(得分:0)

len(a)的值为2,FOR计数从0开始,因此,请求数组中的第3个元素,您必须执行for i in range(len(a) - 1):

答案 1 :(得分:0)

当您尝试将其与负数一起使用时,

sqrt会出现该错误。 sqrt(-16)给出了该错误,因为结果是一个复数。

from math import sqrt
a=[[5, 10], [16, 23]]


b=[[6, 1], [21, 2]]

dL=[]
dist=[]
k=0
h=1
for i in range(len(a)):
    dL=[]
    for j in range(len(a)):
        temp1 = (a[i][k]-b[j][k]) ** 2
        temp2 = (a[i][h]-b[j][h]) ** 2
        temp = temp1 + temp2
        if temp > 0:
            d = sqrt(temp)

        dL.append(d)
    dist.append(dL)

答案 2 :(得分:0)

正如@not_a_robot在他们的评论中提到的,Python中的取幂运算符是**而不是^,因为它在(许多)其他编程语言中。您可以通过简单地替换它们来修复它。

import math

a= [[ 5, 10],
    [16, 23]]
b= [[ 6,  1],
    [21,  2]]

dL=[]
dist=[]
k=0
h=1
for i in range(len(a)):
    dL=[]
    for j in range(len(a)):
        d = math.sqrt((a[i][k]-b[j][k])**2 + (a[i][h]-b[j][h])**2)
        dL.append(d)
    dist.append(dL)

正如我在评论中提到的,最好使用math.hypot(),因为它可以更快地为您提供相同的结果,并且更易于阅读(和写入):

for i in range(len(a)):
    dL=[]
    for j in range(len(a)):
        d = math.hypot(a[i][k]-b[j][k], a[i][h]-b[j][h])
        dL.append(d)
    dist.append(dL)
相关问题