# scoding=utf-8import pylab as plpoints = [[int(eachpoint.split("#")[0]), int(eachpoint.split("#")[1])] for eachpoint in open("points","r")]# 指定三個(gè)初始質(zhì)心currentCenter1 = [20,190]; currentCenter2 = [120,90]; currentCenter3 = [170,140]pl.plot([currentCenter1[0]], [currentCenter1[1]],'ok')pl.plot([currentCenter2[0]], [currentCenter2[1]],'ok')pl.plot([currentCenter3[0]], [currentCenter3[1]],'ok')# 記錄每次迭代后每個(gè)簇的質(zhì)心的更新軌跡center1 = [currentCenter1]; center2 = [currentCenter2]; center3 = [currentCenter3]# 三個(gè)簇group1 = []; group2 = []; group3 = []for runtime in range(50):  group1 = []; group2 = []; group3 = []  for eachpoint in points:    # 計(jì)算每個(gè)點(diǎn)到三個(gè)質(zhì)心的距離    distance1 = pow(abs(eachpoint[0]-currentCenter1[0]),2) + pow(abs(eachpoint[1]-currentCenter1[1]),2)    distance2 = pow(abs(eachpoint[0]-currentCenter2[0]),2) + pow(abs(eachpoint[1]-currentCenter2[1]),2)    distance3 = pow(abs(eachpoint[0]-currentCenter3[0]),2) + pow(abs(eachpoint[1]-currentCenter3[1]),2)    # 將該點(diǎn)指派到離它最近的質(zhì)心所在的簇    mindis = min(distance1,distance2,distance3)    if(mindis == distance1):      group1.append(eachpoint)    elif(mindis == distance2):      group2.append(eachpoint)    else:      group3.append(eachpoint)  # 指派完所有的點(diǎn)后，更新每個(gè)簇的質(zhì)心  currentCenter1 = [sum([eachpoint[0] for eachpoint in group1])/len(group1),sum([eachpoint[1] for eachpoint in group1])/len(group1)]  currentCenter2 = [sum([eachpoint[0] for eachpoint in group2])/len(group2),sum([eachpoint[1] for eachpoint in group2])/len(group2)]  currentCenter3 = [sum([eachpoint[0] for eachpoint in group3])/len(group3),sum([eachpoint[1] for eachpoint in group3])/len(group3)]  # 記錄該次對(duì)質(zhì)心的更新  center1.append(currentCenter1)  center2.append(currentCenter2)  center3.append(currentCenter3)# 打印所有的點(diǎn)，用顏色標(biāo)識(shí)該點(diǎn)所屬的簇pl.plot([eachpoint[0] for eachpoint in group1], [eachpoint[1] for eachpoint in group1], 'or')pl.plot([eachpoint[0] for eachpoint in group2], [eachpoint[1] for eachpoint in group2], 'oy')pl.plot([eachpoint[0] for eachpoint in group3], [eachpoint[1] for eachpoint in group3], 'og')# 打印每個(gè)簇的質(zhì)心的更新軌跡for center in [center1,center2,center3]:  pl.plot([eachcenter[0] for eachcenter in center], [eachcenter[1] for eachcenter in center],'k')pl.show()