自己实现的kmeans的python版本