Как ускорить получение очков на расстоянии, используя два DataFrame? ⇐ Python

[Anonymous]

У меня есть два DataFrame (df и location_df), и оба имеют значения долготы и широты. Я пытаюсь найти точки df в пределах 2 км от каждой строки location_df.
Я пытался векторизовать функцию, но скорость все равно низкая, когда location_df является большой DataFrame (nrows>1000). Есть идеи, как ускориться?
import pandas as pd
import numpy as np

def select_points_for_multiple_locations_vectorized(df, locations_df, radius_km):
R = 6371 # Earth's radius in kilometers

# Convert degrees to radians
df_lat_rad = np.radians(df['latitude'].values)[:, np.newaxis]
df_lon_rad = np.radians(df['longitude'].values)[:, np.newaxis]
loc_lat_rad = np.radians(locations_df['lat'].values)
loc_lon_rad = np.radians(locations_df['lon'].values)

# Haversine formula (vectorized)
dlat = df_lat_rad - loc_lat_rad
dlon = df_lon_rad - loc_lon_rad
a = np.sin(dlat/2)**2 + np.cos(df_lat_rad) * np.cos(loc_lat_rad) * np.sin(dlon/2)**2
c = 2 * np.arctan2(np.sqrt(a), np.sqrt(1-a))
distances = R * c

# Create a mask for points within the radius
mask = distances

Подробнее здесь: https://stackoverflow.com/questions/790 ... dataframes