[已解决]不知道哪里出了问题

jody970214

想用Python自己实现一下K-mean算法，代码如下：

1. def K_means(data, k = 10, iters = 100):
   
2.     center = list(init_center(data, k))
   
3.     while iters:
   
4.         clusters = dict.fromkeys(list(range(k)),[])
   
5.         for node in data:
   
6.             print(node)
   
7.             jud = []
   
8.             center_new = []
   
9.             dist = []
   
10.             for cen in center:
    
11.                 dist.append(np.linalg.norm(node - cen))
    
12.             print(dist)
    
13.             ##cluster = dist.index(min(dist))
    
14.             ##print(cluster)
    
15.             ##clusters[cluster].append(node)
    
16.             ##print(clusters)
    
17.         for cluster in clusters.keys():
    
18.             center_new.append(np.mean(clusters[cluster],axis = 0))  
    
19.         i = k
    
20.         while i:
    
21.             jud.append(all(center_new[i-1] == center[i-1]))
    
22.             i -= 1
    
23.         if all(jud):
    
24.             break
    
25.         center = center_new
    
26.         print(center)
    
27.         iters -= 1
    
28.     return center

复制代码

然后使用

1. test = np.array([[0,0],[1,1],[5,3],[10,1],[0,1]])

复制代码

测试，发现#标部分代码没按想象中的实现，结果如下：

[0 0][1.0, 0.0, 10.04987562112089]
1
{0: [array([0, 0])], 1: [array([0, 0])], 2: [array([0, 0])]}

为什么 clusters[cluster].append(node) 将值放入了所有键下面啊？不应该只放在clusters[1]下面吗？

最佳答案

月排行榜 / 总排行榜

抉择啊

2020-9-22 21:19:44

抉择啊 发表于 2020-9-23 11:44
原因：

补充：覆盖原本的值

1. dict1 = dict.fromkeys(list(range(4)))
   
2. for k in dict1:
   
3.     dict1[k] = []
   
4. 
   
5. dict1[1].append('123')
   
6. 
   
7. for k in range(4):
   
8.     print(k,dict1[k],id(dict1[k]))

复制代码

跳转到最佳答案楼层

抉择啊

抉择啊 发表于 2020-9-23 11:44
原因：

补充：覆盖原本的值

1. dict1 = dict.fromkeys(list(range(4)))
   
2. for k in dict1:
   
3.     dict1[k] = []
   
4. 
   
5. dict1[1].append('123')
   
6. 
   
7. for k in range(4):
   
8.     print(k,dict1[k],id(dict1[k]))

复制代码

jody970214

顶顶顶！！

抉择啊

原因：

1. for k in range(4):
   
2.     clusters = dict.fromkeys(list(range(k)),[]) # 这里的用的都是同一个内存
   
3. 
   
4. print(clusters)
   
5. clusters[1].append('123')
   
6. # clusters.update({1:'123'}) 用这个就不会
   
7. print(clusters)

复制代码

应该还有很多别的解决方法，具体问题应该就是这个，楼主自己在找找好的解决方法。