时间戳

首页 > 解决方案 > 为数据点生成“K”最近邻

python - 为数据点生成“K”最近邻

问题描述

给定一个数据点，我需要生成 K 个最近邻。我阅读了 sklearn 的sklearn.neighbours 模块，但它会在两组数据之间生成邻居。我想要的可能是最接近传递的数据点的 100 个数据点的列表。

无论如何，任何 KNN 算法都应在后台找到这些 K 数据点。有什么办法可以将这些 K 点作为输出返回？

这是我的示例笔记本。

标签： pythonpandasscikit-learnknnnearest-neighbor

解决方案

from sklearn.neighbors import NearestNeighbors

这可以为您提供数据集中 k 个最近邻居的索引。使用kneighbors，第一个值是距离，第二个值是邻居的索引。从文档：

>>> samples = [[0., 0., 0.], [0., .5, 0.], [1., 1., .5]]
>>> from sklearn.neighbors import NearestNeighbors
>>> neigh = NearestNeighbors(n_neighbors=1)
>>> neigh.fit(samples) 
NearestNeighbors(algorithm='auto', leaf_size=30, ...)
>>> print(neigh.kneighbors([[1., 1., 1.]])) 
(array([[0.5]]), array([[2]]))

推荐阅读