时间戳

首页 > 解决方案 > 如何摆脱使用代码查找最近邻居时遇到的此错误？

python - 如何摆脱使用代码查找最近邻居时遇到的此错误？

问题描述

我编写了以下函数，它获取数据、要查找的对等点数和索引以查找前 N 个最近邻：

def fit_nearest_neighbors(data, number_of_peers, index):
    peer_data = FindPeers.filt_data(data)
    peer_data_array = np.array(peer_data)
    knn = NearestNeighbors(algorithm = 'auto', n_neighbors = number_of_peers, metric = 'minkowski', p = 2)
    knn.fit(peer_data_array)
    return knn.kneighbors(peer_data_array[index], return_distance = False)

但是我在 return 后的最后一行代码中收到以下错误：

ValueError: Expected 2D array, got 1D array instead:
array=[2.86839521e-01 7.63588709e-01 1.00000000e+00 1.73483898e-01
 0.00000000e+00 1.25068828e-02 1.66424454e-17 4.38357126e-01
 7.55219585e-03 6.03820534e-02 2.72387749e-01].
Reshape your data either using array.reshape(-1, 1) if your data has a single feature or array.reshape(1, -1) if it contains a single sample.

它打印的错误数组基本上是那个特定的数据点index。我理解这个错误，但我不明白如何使它正确。任何类型的线索/答案都会有所帮助。

标签： pythonnumpymachine-learningdata-scienceknn

解决方案

错误信息很清楚。您需要使用 numpy 函数将数据从 1D 重塑为 2D np.reshape()。

你可以做到这一点

peer_data_array = np.array(peer_data).reshape(-1, 1)

推荐阅读