KMeans implementation issue

KMeans implementation issue
```python
centers = np.array([X[clusters == c].mean(0) for c in clusters])
```
centers's shape is not right, the correct implementation is suggested as
```python
centers = np.array([X[clusters == c].mean(0) for c in range(len(centers))])
```