擅长:python、mysql、java
<p>我知道我可能会迟到,但这是我解决你问题的一般办法:</p>
<pre><code>def predict(data, centroids):
centroids, data = np.array(centroids), np.array(data)
distances = []
for unit in data:
for center in centroids:
distances.append(np.sum((unit - center) ** 2))
distances = np.reshape(distances, data.shape)
closest_centroid = [np.argmin(dist) for dist in distances]
print(closest_centroid)
</code></pre>