【发布时间】:2021-11-01 07:13:14
【问题描述】:
我写了一个函数来显示肘部来选择 Kmeans 的 K 的最佳值。
from sklearn.cluster import KMeans
def show_elbow(df):
distance_list=[]
K = range(1,9)
for k in K:
kmeams=KMeans(n_cluster=k,max_iter=100)
kmeans=KMeans.fit(df)
distance_list.append(kmeans.inertia_)
plt.plot(K,distance_list,'bx-')
当我使用这个函数时,df[''] 会导致错误,而df[['']] 不会。不知道是什么原因。
【问题讨论】: