Как включить высоту в евклидову матрицу расстояний в пандах? - PullRequest
Новая
       34

Как включить высоту в евклидову матрицу расстояний в пандах?

0 голосов
/ 14 мая 2019

У меня есть следующие dataframe в пандах: 

import pandas as pd

df = pd.DataFrame({
    "CityId": {
        "0": 0, 
        "1": 1, 
        "2": 2, 
        "3": 3, 
        "4": 4
    }, 
    "X": {
        "0": 316.83673906150904, 
        "1": 4377.40597216624, 
        "2": 3454.15819771172, 
        "3": 4688.099297634771, 
        "4": 1010.6969517482901
    }, 
    "elevation_meters": {
        "0": 1, 
        "1": 2, 
        "2": 3, 
        "3": 4, 
        "4": 5
    }, 
    "Y": {
        "0": 2202.34070733524, 
        "1": 336.602082171235, 
        "2": 2820.0530112481106, 
        "3": 2935.89805580997, 
        "4": 3236.75098902635
    }
})

Я пытаюсь создать матрицу расстояний, которая представляет стоимость перемещения между каждым из этих CityIds. Используя pdist и squareform из scipy.spatial.distance, я могу сделать следующее: 

from scipy.spatial.distance import pdist, squareform

df_m = pd.DataFrame(
    squareform(
        pdist(
            df[['CityId', 'X', 'Y']].iloc[:, 1:],
            metric='euclidean')
    ),
    index=df.CityId.unique(),
    columns= df.CityId.unique()
)

Это дает мне матрицу расстояний между всеми CityIds, используя попарные расстояния, вычисленные из pdist.

Я хотел бы включить elevation_meters в эту матрицу расстояний. Какой эффективный способ сделать это?

1 Ответ

1 голос
/ 14 мая 2019

Вы можете попробовать scipy.spatial.distance_matrix: 

xx = df[['X','elevation_meters', 'Y']]
pd.DataFrame(distance_matrix(xx,xx), columns= df['CityId'],
             index=df['CityId'])

Выход: 

CityId  0               1                2              3               4
CityId                  
0       0.000000        4468.691544     3197.555070     4432.386687     1245.577226
1       4468.691544     0.000000        2649.512402     2617.799439     4443.602402
2       3197.555070     2649.512402     0.000000        1239.367465     2478.738402
3       4432.386687     2617.799439     1239.367465     0.000000        3689.688537
4       1245.577226     4443.602402     2478.738402     3689.688537     0.000000
...