# -*- coding: utf-8 -*-
"""
@Description:
@Author: gaofan
@Date: 2020-09-23 16:46
"""
import numpy as np
from scipy.spatial.distance import pdist, squareform
# 方法一:矩阵运算
input_array = np.array([[1, 2], [3, 4], [5, 6]])
input_array2 = input_array * input_array
input_array2s = input_array2.sum(axis=1)
input_array2ss = input_array2s.reshape(3, 1)
print(np.sqrt((input_array2ss+input_array2s)-(2 * np.dot(input_array, input_array.T))))
# 方法二:循环赋值
output_array = np.array([[0.] * input_array.shape[0]] * input_array.shape[0])
# 循环输出数组
for i in range(0, output_array.shape[0]):
for j in range(0, output_array.shape[1]):
output_array[i][j] = np.sqrt(pow((input_array[i][0] - input_array[j][0]), 2) + pow(
(input_array[i][1] - input_array[j][1]), 2))
print(output_array.astype(np.float64))
# 方法三:函数
d = pdist(input_array)
output_array = squareform(d)
print(output_array)
计算矩阵行之间的欧氏距离作为新矩阵的元素
最新推荐文章于 2021-08-25 01:58:58 发布