时间戳

首页 > 解决方案 > 有没有一种简单的方法可以计算python中整个数据集在3D空间中两点的距离？

python - 有没有一种简单的方法可以计算python中整个数据集在3D空间中两点的距离？

问题描述

我是 python 新手，我想快速而肮脏地创建一个包含两个坐标（它们的 x、y、z 值）及其相互距离的表。

数据集如下所示

data = np.random.randint(5,30,size=(10,6))
df = pd.DataFrame(data, columns=['x1', 'y1', 'z1', 'x2', 'y2', 'z2'])

print(df)

对于距离列，我想应用一个简单的功能，例如

def distance(x1, y1, z1, x2, y2, z2):
    
    d = math.sqrt(math.pow(x2 - x1, 2) +
                math.pow(y2 - y1, 2) +
                math.pow(z2 - z1, 2)* 1.0)
    return d

我也尝试使用数组（a和b）作为输入，但我还不明白如何将数组作为函数中的变量

p1 = a
p2 = b

squared_dist = np.sum((p1-p2)**2, axis=0)
dist_ab = np.sqrt(squared_dist)
print(dist_ab)

你们将如何处理这项任务？我希望这很容易理解，因为我认为我在过去的几个小时里迷路并且发疯了：D

标签： pythonarraysdistance

解决方案

像这样的东西：

df['distance'] = df.apply(lambda r: distance(r[0], r[1], r[2], r[3], r[4], r[5]), axis=1)

推荐阅读