求最短距离

frankzhang57

想写一个方程，def (place,k,locations):
place是一个坐标,
locations 是一个列表包括地点和坐标 [("旅店", 4.4, 0), ("车站", 5, 3.5), ("广场,4,0")]

如果任意输入一个place (x,y), k是返回地点名称的个数。要求距离从小到大排列，也就是(x,y) 到哪个地点的距离最近，利用 根号(x1-x2)^2 + (y1-y2)^2 来求出locations里面每个点与(x,y)的距离。如果k =1, 则显示最近的一个地点，k=2，则显示最近的两个地点，顺序按着距离从小到大排列。如有等距离，则显示第一个出现的最短地点。

例如 place =(0,0), k =1, 最终显示 ["广场"]
        place =(0,0), k =2, 最终显示["广场"，“旅店”]

小甲鱼

1. 使用公式“号(x1-x2)^2 + (y1-y2)^2”求出 place 参数与 locations 参数中各个元素的距离

2. 按从小到大的次序重新排序

3. 根据 k 的值返回相关的结果

frankzhang57

小甲鱼 发表于 2017-8-31 10:19
1. 使用公式“号(x1-x2)^2 + (y1-y2)^2”求出 place 参数与 locations 参数中各个元素的距离

2. 按从小 ...

老师，我知道是这个思路，但是代码试不出来。我现在的思路是：

def kclosest(place,k,locations):
    location_list = []
    dist_list = []
    # calculate the distances between place and each location in locations:
    for each in locations:  
        dist = ((place[0]-each[1])**2+(place[1]-each[2])**2)**0.5
        dist_list.extend((each[0],dist))
   
    #find the minimum value of distance in the list, then find the location that
    #corresponds with the distance
    for i in range(len(dist_list)):
        for j in range(len(dist_list)):
            if dist_list[i][1] == min(dist_list):
                location_list.append[dist_list[i][0]]
                dist_list.pop[i]
    #return the first k element, so we need k+1 exclusive
    return dist_list[:k+1][0]

我不确定如何按着距离排列以后，再把对应的地点也写出来。index对不上
谢谢