fix kmeans, use ndarray instead of expr.zeros