SQLAlchemy中能否基于自定义距离函数排序查询结果？

你这么写不行，原因很简单：SQLAlchemy的order_by方法是用来生成SQL语句的，最终会把排序逻辑交给数据库执行，但你定义的getdistance是个Python函数，数据库根本不知道这个函数是什么，执行时会直接抛出SQL语法错误。

下面给你两种可行的解决方案：

方案一：小数据量场景——Python端计算排序 #

如果你的数据库里邮编数据不多，直接把所有数据查出来，在Python里计算距离后排序就行：

# 先查出所有数据
all_records = database.query(DataBase).all()
# 按距离排序，key参数指定用getdistance计算的结果作为排序依据
sorted_neighbors = sorted(all_records, key=lambda record: getdistance(input_zipcode, record.zipcode))

方案二：大数据量场景——数据库端计算排序 #

如果数据量很大，方案一的效率会很低，建议把距离计算逻辑搬到数据库端：

1. 先给你的DataBase表新增两个字段：latitude（纬度）和longitude（经度），类型用浮点型就行。
2. 提前用pgeocode把每个邮编对应的经纬度查出来，批量更新到这两个字段里。
3. 然后用数据库支持的地理函数计算距离并排序（以PostgreSQL为例，用Haversine公式计算球面距离）：

from sqlalchemy import func
import pgeocode

# 先获取输入邮编的经纬度
nomi = pgeocode.Nominatim('us')  # 这里替换成你需要的国家代码
input_info = nomi.query_postal_code(input_zipcode)
input_lat, input_lon = input_info.latitude, input_info.longitude

# 用Haversine公式计算距离并排序，6371是地球半径（公里）
neighbors = database.query(DataBase).order_by(
    func.acos(
        func.sin(func.radians(input_lat)) * func.sin(func.radians(DataBase.latitude)) +
        func.cos(func.radians(input_lat)) * func.cos(func.radians(DataBase.latitude)) *
        func.cos(func.radians(input_lon) - func.radians(DataBase.longitude))
    ) * 6371
)

如果你的数据库支持地理扩展（比如PostgreSQL的PostGIS），可以用更简洁的ST_Distance函数来计算距离，效率也更高。

内容的提问来源于stack exchange，提问作者takeshi ookane