Python机器学习——KNN算法

一、底层算法

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

#先随机设置十个样本点表示十杯酒
rowdata = {
   '颜色深度':[14.23,13.2,13.16,14.37,13.24,12.07,12.43,11.79,12.37,12.04],
          '酒精浓度':[5.64,4.38,5.68,4.80,4.32,2.76,3.94,3.  ,2.12,2.6 ],
          '品种':[0,0,0,0,0,1,1,1,1,1]}
# 0代表黑皮诺 ， 1代表赤霞珠
wine_data = pd.DataFrame(rowdata)
wine_data

X = np.array(wine_data.iloc[:,0:2])
X
y = np.array(wine_data.iloc[:,-1])
y
new_data = np.array([12.8,4.1])   #要判断的数据

# 1 算距离
from math import sqrt
distance = [sqrt(np