KNN-预测单个样本

def solve():
    import pandas as pd
    import numpy as np
    from sklearn.neighbors import KNeighborsClassifier
    df = pd.read_csv('https://liangdaima.com/static/data/iris.csv')
    X = df.drop('species', axis=1)
    y = df['species']
    knn = KNeighborsClassifier(n_neighbors=3)
    knn.fit(X, y)
    sample = np.array([[5.1, 3.5, 1.4, 0.2]])
    pred = knn.predict(sample)
    print('预测类别:', pred[0])