Расстояние между точками в двух фреймах данных - PullRequest
Новая
       47

Расстояние между точками в двух фреймах данных

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

Я новичок в python, и я попытался найти, но не смог найти решение.

У меня есть два кадра данных с декартовыми координатами. 

node    x      y    value
abc    645    714   8 
def    187    754   11 

location  x    y    value
ijk      621  744   1 
lmn      202  720   -5

Я хочу найти минимальное расстояние каждого местоположения в df2 с узлами в df1 и обновить «значение» в df1.

Я видел cdist для вычисления минимального расстояния, но как мне связать это с соответствующим? узел

Поскольку ijk близко к ab c и lmn к def, окончательный ответ должен быть 

node    x      y    value
abc    645    714   9 
def    187    754   6

1 Ответ

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

Вот способ сделать это с нуля: 

import numpy as np
import pandas as pd

df1 = pd.DataFrame({'node': ['abc', 'def'], 
                    'x': [645, 187], 
                    'y': [714, 754], 
                    'value': [8, 11]}) 
df2 = pd.DataFrame({'location': ['ijk', 'lmn'], 
                    'x': [621,  202], 
                    'y': [744, 720], 
                    'value': [1, -5]}) 

# compute Euclidean distances (using Pythagoras' theorem)
df2['distances'] = [[((df1['x'][j] - df2['x'][i]) ** 2 
                    + (df1['y'][j] - df2['y'][i]) ** 2) ** 0.5 
                     for j in range(len(df1))] 
                    for i in range(len(df2))]
df2

    location  x     y      value  distances
0   ijk       621   744     1     [38.41874542459709, 434.11519208615584]
1   lmn       202   720    -5     [443.0406301909566, 37.16180835212409]

df1['new_value'] = [df1['value'][row] + df2['value'][np.argmin(df2['distances'][row])] 
                    for row in range(len(df1))] 
df1

    node    x       y      value  new_value
0   abc     645     714     8     9
1   def     187     754    11     6
...