6262
6363colors = "brgy"
6464fig , ax = plt .subplots (1 , 1 , figsize = (4 , 4 ))
65- for i in range (0 , max (cl ) + 1 ):
65+ for i in range (max (cl ) + 1 ):
6666 ax .plot (X [cl == i , 0 ], X [cl == i , 1 ], colors [i ] + "." , label = "cl%d" % i )
6767 x = [km .cluster_centers_ [i , 0 ], km .cluster_centers_ [i , 0 ]]
6868 y = [km .cluster_centers_ [i , 1 ], km .cluster_centers_ [i , 1 ]]
9797#
9898
9999fig , ax = plt .subplots (1 , 2 , figsize = (10 , 4 ))
100- for i in range (0 , max (cl1 ) + 1 ):
100+ for i in range (max (cl1 ) + 1 ):
101101 ax [0 ].plot (X [cl1 == i , 0 ], X [cl1 == i , 1 ], colors [i ] + "." , label = "cl%d" % i )
102102 ax [1 ].plot (X [cl2 == i , 0 ], X [cl2 == i , 1 ], colors [i ] + "." , label = "cl%d" % i )
103103 x = [km1 .cluster_centers_ [i , 0 ], km1 .cluster_centers_ [i , 0 ]]
@@ -135,7 +135,7 @@ def plot_delaunay(ax, edges, points):
135135
136136
137137fig , ax = plt .subplots (1 , 2 , figsize = (10 , 4 ))
138- for i in range (0 , max (cl ) + 1 ):
138+ for i in range (max (cl ) + 1 ):
139139 ax [0 ].plot (X [cl == i , 0 ], X [cl == i , 1 ], colors [i ] + "." , label = "cl%d" % i )
140140 x = [km .cluster_centers_ [i , 0 ], km .cluster_centers_ [i , 0 ]]
141141 y = [km .cluster_centers_ [i , 1 ], km .cluster_centers_ [i , 1 ]]
@@ -145,7 +145,7 @@ def plot_delaunay(ax, edges, points):
145145
146146cls = km .cluster_centers_iter_
147147ax [1 ].plot (X [:, 0 ], X [:, 1 ], "." , label = "X" , color = "#AAAAAA" , ms = 3 )
148- for i in range (0 , max (cl ) + 1 ):
148+ for i in range (max (cl ) + 1 ):
149149 ms = numpy .arange (cls .shape [- 1 ]).astype (numpy .float64 ) / cls .shape [- 1 ] * 50 + 1
150150 ax [1 ].scatter (cls [i , 0 , :], cls [i , 1 , :], color = colors [i ], s = ms , label = "cl%d" % i )
151151 plot_delaunay (ax [1 ], edges , km .cluster_centers_ )
0 commit comments