如何去除scikit-learn KDTree.query_radius()空输出并获取唯一值

可以直接利用NumPy的内置方法，无需循环实现需求：

import numpy as np
from sklearn.neighbors import KDTree

# 初始化KDTree并执行半径查询
kd_tree = KDTree(X)
idxs = kd_tree.query_radius(Y, r=0.2)

# 扁平化结果、去重并得到排序后的唯一索引
unique_sorted_idxs = np.unique(np.concatenate(idxs))

关键说明 #

• np.concatenate(idxs)：将query_radius返回的索引列表直接合并为一维数组，自动忽略其中的空数组，无需手动遍历处理。
• np.unique()：默认返回排序后的唯一元素数组，完全匹配你需要的输出格式。

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