def get_sse(data, centroid, q):
sse = []
for i in range(len(centroid)):
sum = 0
for j in range(len(q)):
if q[j] == i:
sum += K_means.get_dist(data[j][0], data[j][1], centroid[i][0], centroid[i][1])
sse.append(sum)
return sse
