diff --git a/pflaenzli/pflaenzli/utils/distance.py b/pflaenzli/pflaenzli/utils/distance.py new file mode 100644 index 0000000..f0c21a8 --- /dev/null +++ b/pflaenzli/pflaenzli/utils/distance.py @@ -0,0 +1,16 @@ +from geopy.distance import distance +import os +from pandas import read_pickle + +path = os.path.dirname(os.path.abspath(__file__)) + +df = read_pickle(os.path.join(path, 'plz.pkl')) + + +def calculate_distance(zip_1, zip_2): + zip_1_coords = tuple(df[df.index == zip_1].values) + zip_2_coords = tuple(df[df.index == zip_2].values) + + dist = distance((zip_1_coords), (zip_2_coords)).kilometers + + return None if dist > 400 else dist diff --git a/pflaenzli/pflaenzli/utils/plz.pkl b/pflaenzli/pflaenzli/utils/plz.pkl new file mode 100644 index 0000000..44cb42d Binary files /dev/null and b/pflaenzli/pflaenzli/utils/plz.pkl differ