Numpy 计算两个数组中各个坐标点之间的间距

引言

今天，遇到了一个问题，有两个数组，其中分别存放着许多坐标点，如何计算两两坐标点之间的欧几里得距离。

正文

关于 np.newaxis 的用法可以参考文章 Python np.newaxis 用法。然后我们可以使用如下代码完成这一操作。

import numpy as np


def euclidean_distance_matrix(A, B):
    """
    计算两个数组中所有点之间的欧式距离矩阵。