Почему моя функция вычисляет неправильное расстояние между двумя точками - PullRequest
Новая
       14

Почему моя функция вычисляет неправильное расстояние между двумя точками

0 голосов
/ 05 ноября 2019

У меня есть датафрейм, который суммирует окрестности и кафе в этих кварталах вместе с их местоположениями, называемый Финал. И генерировал случайные точки местоположения, каждые 100 метров друг от друга. Теперь я хотел бы рассчитать расстояние между точками данных (кафе) в окончательном фрейме данных и ближайшим местоположением, которое было сгенерировано случайным образом.

Я пытаюсь клонировать код в этой записной книжке (кодовый блок 92-96)

Я думаю, что проблема в моем исходном кадре данных. Но я не знаю, что мне следует делать с фреймом данных, чтобы эта работа работала.

Переменная events в приведенном ниже коде хранит итоговый фрейм данных, превращенный в словарь (потому что в коде, который я пытаюсь клонировать, они использовали словарь). Однако в моей функции find_nearest_restaurant res [4] не принимает широту и долготу ... Не уверен, что с ним делать.

Конечный фрейм данных Вот так выглядит фрейм данных

Я пытаюсь следующий код для этого: 

def find_nearest_restaurant(x, y, events):
    d_min = 100000
    for res in events:
        res_x = res[4]; res_y = res[5]
        d = calc_xy_distance(x, y, res_x, res_y)
        if d<=d_min:
            d_min = d
    return d_min


def lonlat_to_xy(lon, lat):
    proj_latlon = pyproj.Proj(proj='latlong',datum='WGS84')
    proj_xy = pyproj.Proj(proj="utm", zone=33, datum='WGS84')
    xy = pyproj.transform(proj_latlon, proj_xy, lon, lat)
    return xy[0], xy[2]


events = Final.to_dict()
Kralingen_center = [51.928263, 4.50344]
Kr_center_x, Kr_center_y = lonlat_to_xy(Kralingen_center[2], Kralingen_center[0]) 
Centrum_Center = [51.922909, 4.47059]
k = math.sqrt(3) / 2 # Vertical offset for hexagonal grid cells
x_step = 100
y_step = 100 * k 
roi_x_min = Kr_center_x - 2000
roi_y_max = Kr_center_y + 1000
roi_center_x = roi_x_min + 2000
roi_center_y = roi_y_max - 2500
roi_y_min = roi_center_y - 2500


def xy_to_lonlat(x, y):
    proj_latlon = pyproj.Proj(proj='latlong',datum='WGS84')
    proj_xy = pyproj.Proj(proj="utm", zone=33, datum='WGS84')
    lonlat = pyproj.transform(proj_xy, proj_latlon, x, y)
    return lonlat[0], lonlat[2]


def calc_xy_distance(x1, y1, x2, y2):
    dx = x2 - x1
    dy = y2 - y1
    return math.sqrt(dx*dx + dy*dy)


roi_latitudes = []
roi_longitudes = []
roi_xs = []
roi_ys = []

for i in range(0, int(51/k)):
    y = roi_y_min + i * y_step
    x_offset = 50 if i%2==0 else 0
    for j in range(0, 51):
        x = roi_x_min + j * x_step + x_offset
        d = calc_xy_distance(Kr_center_x, Kr_center_y, x, y)
        if (d <= 2001):
            lon, lat = xy_to_lonlat(x, y)
            roi_latitudes.append(lat)
            roi_longitudes.append(lon)
            roi_xs.append(x)
            roi_ys.append(y)


print(len(roi_latitudes), 'candidate points generated.')


def find_nearest_restaurant(x, y, events):
    d_min = 100000
    for res in events:
        res_x = res[4]; res_y = res[5]
        d = calc_xy_distance(x, y, res_x, res_y)
        if d<=d_min:
            d_min = d
    return d_min

Я хотел бы получить кадр данных с: 

df_roi_locations = pd.DataFrame({'Latitude':roi_latitudes,
                                 'Longitude':roi_longitudes,
                                 'X':roi_xs,
                                 'Y':roi_ys,
                                 'Distance to restaurant':roi_kr_distances})

roi_kr_distances это просто результат выполнения *Функция 1022 * и добавление ее в пустой список.

Но код выдает ошибку: 

<ipython-input-70-b6a8fa5871aa> in find_nearest_restaurant(x, y, events)
     53     for res in events:
     54         res_x = res[3]; res_y = res[4]
---> 55         d = calc_xy_distance(x, y, res_x, res_y)
     56         if d<=d_min:
     57             d_min = d

<ipython-input-70-b6a8fa5871aa> in calc_xy_distance(x1, y1, x2, y2)
     26 
     27 def calc_xy_distance(x1, y1, x2, y2):
---> 28     dx = x2 - x1
     29     dy = y2 - y1
     30     return math.sqrt(dx*dx + dy*dy)

TypeError: unsupported operand type(s) for -: 'str' and 'float'

1 Ответ

0 голосов
/ 05 ноября 2019

Трудно сказать из приведенного вами примера кода, потому что я не знаю, что хранится внутри переменной events, но из вашей ошибки просто выглядит, что у вас есть два типа данных (string и Float)и вы пытаетесь выполнить на них вычитание, которое не разрешено в python.

Это единственная попытка предложить, то есть попытка преобразовать ваши параметры в числа с плавающей точкой. 

def calc_xy_distance(x1, y1, x2, y2):
    dx = float(x2) - float(x1)
    dy = float(y2) - float(y1)
    return math.sqrt(dx*dx + dy*dy)

Надеюсь, это поможет.

...