在python中计算间距不同于指定纬度/经度网格的渐变

import numpy as np
import math

start_lat = 40.7128  # 起始点纬度
start_lon = -74.0060  # 起始点经度
end_lat = 34.0522  # 结束点纬度
end_lon = -118.2437  # 结束点经度

def calculate_distance(lat1, lon1, lat2, lon2):
    R = 6371  # 地球平均半径（单位：公里）
    dlat = math.radians(lat2 - lat1)
    dlon = math.radians(lon2 - lon1)
    a = math.sin(dlat / 2) * math.sin(dlat / 2) + math.cos(math.radians(lat1)) * math.cos(math.radians(lat2)) * math.sin(dlon / 2) * math.sin(dlon / 2)
    c = 2 * math.atan2(math.sqrt(a), math.sqrt(1 - a))
    distance = R * c
    return distance

distance = calculate_distance(start_lat, start_lon, end_lat, end_lon)

num_rows = 10  # 网格行数
num_cols = 10  # 网格列数

lat_distance = (end_lat - start_lat) / num_rows
lon_distance = (end_lon - start_lon) / num_cols

grid = np.zeros((num_rows, num_cols, 2))  # 二维数组，每个元素包含纬度和经度

for i in range(num_rows):
    for j in range(num_cols):
        grid[i][j][0] = start_lat + (i * lat_distance)
        grid[i][j][1] = start_lon + (j * lon_distance)