我有以下代码来计算地图上各点之间的距离。我的目标是做到以下几点:
从位置词典
循环通过字典并计算与所有人的距离 指向起点。
找到距离起点最小距离的位置。
然后将该位置与起点配对以形成一对节点 连接边缘。
执行此操作后,将最小位置作为新起点 并从位置字典中删除它。
然后我会回到上一步,其余的要点。
我目前能够从第一个起始点获得距离,但无法遍历位置字典中的其余点。
非常感谢任何建议。
from math import atan2, cos, sin, sqrt, radians
start = (43.82846160000000000000, -79.53560419999997000000)
locations = {
'one':(43.65162010000000000000, -79.73558579999997000000),
'two':(43.75846240000000000000, -79.22252100000003000000),
'thr':(43.71773540000000000000, -79.74897190000002000000)
}
cal_distances = {}
nodes = []
def dis():
y = len(locations)
x = 0
while x != y:
for key, value in locations.iteritems():
d = calc_distance(value)
cal_distances.setdefault(key,[])
cal_distances[key].append(d)
print cal_distances
min_distance = min(cal_distances, key = cal_distances.get)
if locations.has_key(min_distance):
for ky, val in locations.iteritems():
if ky == min_distance:
start = val
locations.pop(ky)
x = x+1
print locations
print nodes
def calc_distance(destination):
"""great-circle distance between two points on a sphere from their longitudes and latitudes"""
lat1, lon1 = start
lat2, lon2 = destination
radius = 6371 # km. earth
dlat = radians(lat2-lat1)
dlon = radians(lon2-lon1)
a = (sin(dlat/2) * sin(dlat/2) + cos(radians(lat1)) * cos(radians(lat2)) * sin(dlon/2) * sin(dlon/2))
c = 2 * atan2(sqrt(a), sqrt(1-a))
d = radius * c
return d
dis()
答案 0 :(得分:0)
您现在的代码非常令人困惑。我想你想要实现的目标是:
start = (43.82846160000000000000, -79.53560419999997000000)
locations = {'one':(43.65162010000000000000, -79.73558579999997000000),
'two':(43.75846240000000000000, -79.22252100000003000000),
'thr':(43.71773540000000000000, -79.74897190000002000000)}
def dis(start, locations):
nodes = []
while locations:
# until the dictionary of locations is empty
nearest = min(locations, key=lambda k: calc_distance(start, locations[k]))
# find the key of the closest location to start
nodes.append((start, locations[nearest]))
# add a tuple (start, closest location) to the node list
start = locations.pop(nearest)
# remove the closest location from locations and assign to start
return nodes
def calc_distance(start, destination):
# ...
nodes = dis(start, locations)
请注意,我已start
明确提出calc_distance
,start
和locations
显式参数dis
- 尽可能不要依赖于访问变量的范围。我在nodes
中得到的输出是:
[((43.8284616, -79.53560419999997), (43.7177354, -79.74897190000002)),
((43.7177354, -79.74897190000002), (43.6516201, -79.73558579999997)),
((43.6516201, -79.73558579999997), (43.7584624, -79.22252100000003))]