1次元配列、2次元配列の変換について

前提・実現したいこと

k近傍の実装をしたいのですがfit()に代入する次元が違うと怒られます

発生している問題・エラーメッセージ

Expected 2D array, got 1D array instead:
array=[294.21104801 -21.56059432].
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.

該当のソースコード

import numpy as np 
import matplotlib.pyplot as plt
from sklearn.neighbors import NearestNeighbors 
import math
with open('scan data 5.txt') as file:
    lines = file.readlines()

x = np.empty((0,2))
for line in lines:
    d = line.split()
    a=float(d[0])*math.pi/180
    t =float(d[1])*math.cos(a)#X
    s =float(d[1])*math.sin(a)#y
    
    X=np.stack((t,s))
    Y=[t,s]
    #print(X)
   # print(5)
    #print(Y)
    #plt.scatter(t, s, marker='.',color='black') 
#plt.show()
k = 1 
test_datapoint = [0, 0] 

knn_model = NearestNeighbors(n_neighbors=k, algorithm='ball_tree').fit(X)
distances, indices = knn_model.kneighbors([test_datapoint]) 

   
plt.title('Nearest neighbors') 
plt.scatter(X[:, 0], X[:, 1], marker='o', s=75, color='k') 
plt.scatter(X[indices][0][:][:, 0], X[indices][0][:][:, 1], 
            marker='o', s=250, color='k', facecolors='none') 
plt.scatter(test_datapoint[0], test_datapoint[1], 
            marker='x', s=75, color='k') 
plt.show()

試したこと

Xの一部の中身が
[-542.04960541 -270.95797695]
[-542.98712474 -277.96579353]
[-544.20491157 -286.47166391]
Yの一部の中身が
[1138.3730167627668, -888.3934233809014]
[1166.7252685949206, -889.2924983502955]
[1195.5014931699388, -889.3121948047528]
です

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

行動規範の内容に同意します

回答1件

ベストアンサー

Python
1for line in lines:
2    d = line.split()
3    a=float(d[0])*math.pi/180
4    t =float(d[1])*math.cos(a)#X
5    s =float(d[1])*math.sin(a)#y
6
7    X=np.stack((t,s))
8    Y=[t,s]
9    #print(X)
10   # print(5)
11    #print(Y)
12    #plt.scatter(t, s, marker='.',color='black') 
13#plt.show()