Kmeans: '(slice(None, None, None), slice(None, None, None))' is an invalid key

https://github.com/marcotcr/lime/issues/293

위에서 나와있듯이

Kmean알고리즘을 그리려고 scatter할때

(slice(None, None, None), slice(None, None, None))' is an invalid key 에러가 발생한다면

iloc을 사용해서 x,y,z 축을 표시해주자

plt.scatter(stratified_data[normal_index].iloc[:,0], stratified_data[normal_index].iloc[:,1],

            km.predict(stratified_data[normal_index]), cmap='viridis')

centers = km.cluster_centers_

plt.scatter(centers[:, 0], centers[:, 1], c='black', alpha=0.5);