python求距离矩阵_Python中给定经纬度数据计算距离矩阵的有效方法

这只是将代码矢量化：def new_get_distances(loc1, loc2):

earth_radius = 3958.75

locs_1 = np.deg2rad(loc1)

locs_2 = np.deg2rad(loc2)

lat_dif = (locs_1[:,0][:,None]/2 - locs_2[:,0]/2)

lon_dif = (locs_1[:,1][:,None]/2 - locs_2[:,1]/2)

np.sin(lat_dif, out=lat_dif)

np.sin(lon_dif, out=lon_dif)

np.power(lat_dif, 2, out=lat_dif)

np.power(lon_dif, 2, out=lon_dif)

lon_dif *= ( np.cos(locs_1[:,0])[:,None] * np.cos(locs_2[:,0]) )

lon_dif += lat_dif

np.arctan2(np.power(lon_dif,.5), np.power(1-lon_dif,.5), out = lon_dif)

lon_dif *= ( 2 * earth_radius )

return lon_dif

locations_1 = np.array([[34, -81], [32, -87], [35, -83]])

locations_2 = np.array([[33, -84], [39, -81], [40, -88], [30, -80]])

old = get_distances(locations_1, locations_2)

new &#