#note: need a knn.csv file in dir
import math
import pandas as pd
c=[]
d=[]
k=5

p = [6,3]
data = pd.read_csv('knn.csv')
len=len(data.values)

cl = [x for x in data['y']]

for i in range(2):
    a = [x for x in data[f'x{i}']]
    c.append(a)

c = [x for x in zip(*c)]

for i in c:
    d.append(math.dist(p,i))

cld = [x for x in zip(d,cl)]
cld.sort()

for i in range(len-k):
    cld.pop()

print(max(cld, key=cld.count))