Typeerror kmeans object is not subscriptable