2D数据点范围查询

Question

我正在处理一个巨大的2d数据集，需要对每个点进行范围查询，将范围内的邻居作为一组返回 我已经使用KD Tree表单sk学习索引进行了测试，但问题是，它将索引作为列表返回，转换为集合需要的时间太长。 是否有数据结构，它将范围查询中的点作为集合而不是列表返回？

Answer 1

结果本身不是列表。

获取k-d-tree的源代码，并进行修改，使其直接写入集合，而不是列表。

但我非常怀疑这会解决你的实际问题。将小列表转换为集合几乎不会成为性能问题......但是，您使用的是python。传统的python set（）比numpy数组慢得多。但是不要因为没有使用慢速设置而责怪数据结构。