在python中假设以下距离矩阵…
0 1 2 3
0 0 1 4 8
1 1 0 3 7
2 4 3 0 3
3 8 7 3 0
我想把这个距离矩阵转换成成对欧几里得距离的列表,如下所示…
Obj1 Obj2 Dist
0 0 1 1
1 0 2 4
2 0 3 8
3 1 2 3
4 1 3 7
5 2 3 3
我似乎找不到任何解决方案,但我是新的python所以也许我只是不知道该搜索什么。如有任何帮助,不胜感激。
distances = [
[0, 1, 4, 8],
[1 ,0, 3, 7],
[4, 3, 0, 3],
[8, 7, 3, 0],
]
MATRIX_SIZE = len(distances)
distance_pairs = []
for i in range(MATRIX_SIZE):
for j in range(i):
distance_pairs.append(("distance from {} to {} is {}".format(i, j, distances[i][j])))
print(distance_pairs)
您可以使用Pandas和DataFrame.stack()
import pandas as pd
import io
txt_dist_mat = ''' 0 1 2 3
0 0 1 4 8
1 1 0 3 7
2 4 3 0 3
3 8 7 3 0'''
df = pd.read_fwf(io.StringIO(txt_dist_mat), index_col=0)
euc_mat = df.stack().reset_index().rename(columns={'level_0': 'Obj1', 'level_1': 'Obj2', 0: 'Dist'})