How can the Euclidean distance be calculated with NumPy?

后端 未结 22 1578
春和景丽
春和景丽 2020-11-22 02:29

I have two points in 3D:

(xa, ya, za)
(xb, yb, zb)

And I want to calculate the distance:

dist = sqrt((xa-xb)^2 + (ya-yb)^2 + (         


        
22条回答
  •  小鲜肉
    小鲜肉 (楼主)
    2020-11-22 02:37

    You can just subtract the vectors and then innerproduct.

    Following your example,

    a = numpy.array((xa, ya, za))
    b = numpy.array((xb, yb, zb))
    
    tmp = a - b
    sum_squared = numpy.dot(tmp.T, tmp)
    result = numpy.sqrt(sum_squared)
    

提交回复
热议问题