метод расчета дистанционной матрицы с использованием numpy вещания - PullRequest
Новая
       9

метод расчета дистанционной матрицы с использованием numpy вещания

0 голосов
/ 22 апреля 2020

Я хочу вычислить евклидову матрицу расстояний (x, y, z) координат. Я нашел способ вычисления евклидовой матрицы расстояний (x, y) координат. 

def euclidian_dist(x,y):
    diff = np.expand_dims(x, axis=1) - np.expand_dims(y, axis=0)
    distance_matrix = np.sqrt(np.sum(diff ** 2, axis=-1))
    return distance_matrix

if __name__=="__main__":
    #[[x1,y1],[x2,y2],...,[xn,yn]]
    p = np.array([[0.0, 0.0],[1., 0],[1, 1],[0, 1]])
    q = np.array([[0., 0],[1., 0],[0.3, 0.4],[0.7, 0.6]])
    euclidian_dist(p,q)

ans: array([[0.        , 1.        , 0.5       , 0.92195445],
            [1.        , 0.        , 0.80622577, 0.67082039],
            [1.41421356, 1.        , 0.92195445, 0.5       ],
            [1.        , 1.41421356, 0.67082039, 0.80622577]])

На следующем шаге я хочу использовать координаты x, y, z ([[x1, y1, z1] , [x2, y2, z2], ..., [xn, yn, zn]])

и хотим получить евклидову матрицу расстояний (форма: n✕n).

Может Вы дадите мне знать, если знаете решение?

1 Ответ

1 голос
/ 22 апреля 2020

Используйте тот же код для 3D-точек. Работает нормально: 

def euclidian_dist(x,y):
    diff = np.expand_dims(x, axis=1) - np.expand_dims(y, axis=0)
    distance_matrix = np.sqrt(np.sum(diff ** 2, axis=-1))
    return distance_matrix

if __name__=="__main__":
    #[[x1,y1],[x2,y2],...,[xn,yn]]
    p = np.array([[0.0, 0.0, 0.0],[1.,1., 0],[1, 1, 1],[0, 0, 1]])
    q = np.array([[0., 0, 0],[1.,1., 0],[0.3,0.3, 0.4],[0.7, 0.6, 0.6]])
    print(euclidian_dist(p,q))

вывод: 

[[0.         1.41421356 0.58309519 1.1       ]
 [1.41421356 0.         1.06770783 0.78102497]
 [1.73205081 1.         1.15758369 0.64031242]
 [1.         1.73205081 0.73484692 1.00498756]]
...