python from milk import kmeans import numpy as np X = np.array([[1, 2], [1, 4], [1, 0], [10, 2], [10, 4], [10, 0]]) centroids, assignments = kmeans(X, k=2) print(centroids) print(assignments)