2014-09-10 3 views
5

나는 내가 하버 사인 기능을 사용할 필요가있는 두 개의 위도, 경도 점 사이의 거리를 찾을 알고 경도입니다. 이 점수가 설정 점 -56.7213600, 37.2175900에서 얼마나 떨어져 있는지 확인하고 싶습니다. DataFrame에서 값을 가져 와서 함수에 넣으려면 어떻게해야합니까?벡터화 하버 사인 공식

예를 들어 DataFrame :

 SEAZ  LAT   LON 
1 296.40, 58.7312210, 28.3774110 
2 274.72, 56.8148320, 31.2923240 
3 192.25, 52.0649880, 35.8018640 
4  34.34, 68.8188750, 67.1933670 
5 271.05, 56.6699880, 31.6880620 
6 131.88, 48.5546220, 49.7827730 
7 350.71, 64.7742720, 31.3953780 
8 214.44, 53.5192920, 33.8458560 
9  1.46, 67.9433740, 38.4842520 
10 273.55, 53.3437310, 4.4716664 
+0

나는 당신의 문제를 해결하는 대답을 올렸지 만, 그것이 최적 이하라고 생각한다. – EdChum

+0

작은 df에 대해서 나의 대답은 괜찮을 것이지만 훨씬 더 큰 df에 대해서는 당신이 라디안으로 변환하고 저장하는 것이 더 낫다. 이 다음 전체 데이터 프레임에 대한 계산을 수행 – EdChum

+0

나는 df에 143 개의 값을 가지고 있으며 이것은 잘 작동 한 것 같다. – user3755536

답변

16

계산이 정확하지만,이 일을 다음 만약 내가 확신 할 수 없습니다 :

In [11]: 

def haversine(row): 
    lon1 = -56.7213600 
    lat1 = 37.2175900 
    lon2 = row['LON'] 
    lat2 = row['LAT'] 
    lon1, lat1, lon2, lat2 = map(radians, [lon1, lat1, lon2, lat2]) 
    dlon = lon2 - lon1 
    dlat = lat2 - lat1 
    a = sin(dlat/2)**2 + cos(lat1) * cos(lat2) * sin(dlon/2)**2 
    c = 2 * arcsin(sqrt(a)) 
    km = 6367 * c 
    return km 

df['distance'] = df.apply(lambda row: haversine(row), axis=1) 
df 
Out[11]: 
     SEAZ  LAT  LON  distance 
index           
1  296.40 58.731221 28.377411 6275.791920 
2  274.72 56.814832 31.292324 6509.727368 
3  192.25 52.064988 35.801864 6990.144378 
4  34.34 68.818875 67.193367 7357.221846 
5  271.05 56.669988 31.688062 6538.047542 
6  131.88 48.554622 49.782773 8036.968198 
7  350.71 64.774272 31.395378 6229.733699 
8  214.44 53.519292 33.845856 6801.670843 
9  1.46 67.943374 38.484252 6418.754323 
10  273.55 53.343731 4.471666 4935.394528 

다음 코드는 작은 dataframe에 실제로 느리지 만 제가 적용 10000 행 df에 :

In [35]: 

%%timeit 
df['LAT_rad'], df['LON_rad'] = np.radians(df['LAT']), np.radians(df['LON']) 
df['dLON'] = df['LON_rad'] - math.radians(-56.7213600) 
df['dLAT'] = df['LAT_rad'] - math.radians(37.2175900) 
df['distance'] = 6367 * 2 * np.arcsin(np.sqrt(np.sin(df['dLAT']/2)**2 + math.cos(math.radians(37.2175900)) * np.cos(df['LAT_rad']) * np.sin(df['dLON']/2)**2)) 

1 loops, best of 3: 17.2 ms per loop 

4.3s 그래서 거의 250 배 빠르고, 뭔가 우리가 한 줄에 모든 위의 압축 할 경우 미래 ​​

에주의해야 :

In [39]: 

%timeit df['distance'] = 6367 * 2 * np.arcsin(np.sqrt(np.sin((np.radians(df['LAT']) - math.radians(37.21759))/2)**2 + math.cos(math.radians(37.21759)) * np.cos(np.radians(df['LAT']) * np.sin((np.radians(df['LON']) - math.radians(-56.72136))/2)**2))) 
100 loops, best of 3: 12.6 ms per loop 

우리는 341 배 빠른 지금 배 ~를 더 속도 업을 관찰합니다.

+0

최적화 비교를 가져 주셔서 감사합니다. 잘 했어. – deepelement