K-Means Clustering in Python - How to Assign Points to Clusters?

Question

I first define this function:

import math

def distance(a, b):
    """Returns Euclidean distance in points (a, b).
    "