I have the following DataFrame in pandas:

I want to create a new column called km_to_fixed_coords in which the distance in coordinates to a new fixed point is calculated, which will be the same for all rows (36.7196, -4.42002).

This code calculates the distance between two coordinates in kilometres.

import geopy.distance
# (latitude, longitude)
coords_1 = (x, y)
fixed_coords = (36.7196, -4.42002)

print(geopy.distance.distance(coords_1, fixed_coords).km)

The resulting DataFrame should look like the example:

CodePudding user response：

from typing import Tuple
import geopy.distance


def distance(
    lat: float, lon: float, fixed_coords: Tuple[float] = (36.7196, -4.42002)
) -> float:
    return geopy.distance.distance((lat, lon), fixed_coords).km


df["km_to_fixed_coords"] = df.apply(lambda row: distance(row.latitude, row.longitude))

CodePudding user response：

apply a method on rows like this ?

import geopy.distance

# (latitude, longitude)
fixed_coords = (36.7196, -4.42002)

df['km_to_fixed_coords'] = df.apply(
    lambda row: geopy.distance.distance((row.latitude, row.longitude), fixed_coords).km,
    axis=1
)