如何获得geodjango的k个最近邻居？

Question

假设我有以下模型：

class Person:
     id       = models.BigAutoField(primary_key=True)
     name     = models.CharField(max_length=150)
     location = models.PointField()

如何使用 geodjango 按位置获取 k 个最近邻 (KNN)？
我必须为此编写自定义 SQL 吗？
我正在将 PostgreSQL 与 PostGIS 一起使用。

Answer 1

您可以使用raw() sql 查询来利用 postgis order_by 运算符：

1. <-> 使用边界框的中心来计算对象间距离。

2. <#> 使用边界框本身来计算对象间距离。

在您的情况下，您想要的似乎是 <-> 运算符，因此是原始查询：

knn = Person.objects.raw(
    'SELECT * FROM myapp_person 
    ORDER BY location <-> ST_SetSRID(ST_MakePoint(%s, %s),4326)',
    [location.x, location.y]
)[:k]

由于自己的愚蠢而编辑：您可以省略 [:k] 以在原始 SQL 查询中添加 LIMIT 1。 （不要像我一样使用两者！）

---

在回答您的另一个问题的过程中：How efficient is it to order by distance (entire table) in geodjango，可能还有另一种解决方案：

通过启用spatial indexing 并通过逻辑约束缩小查询范围（如上述链接问题的in my answer 所述），您可以实现非常快速的KNN 查询，如下所示：

current_location = me.location
people = People.objects.filter(
    location__dwithin=(current_location, D(km=50))
).annotate(
    distance=Distance('location', current_location)
).order_by('distance')[:k]