如何指定列以使用Scikit Learn构建KNN分类器

Question

我的数据有一个多维数组。我只想根据其中一列查找最近的邻居，但希望能够读取其他列中的值。如何指定要使用的列？

Answer 1

>>> import numpy as np
>>> A = np.array([[1,2,3,4],[5,6,7,8]])
>>> A
array([[1, 2, 3, 4],
       [5, 6, 7, 8]])
>>> A[:,2] # returns the third column
array([3, 7])

通常，使用array [：，n-1]返回多维数组的第n列。