StatRed ass4: Source code cleanup.

statred/ass4/k-means.py

-from pylab import loadtxt, array, scatter, figure, show, mean, argmin, \
-        append, savefig
+from pylab import loadtxt, array, scatter, figure, show, mean, argmin, append
 from random import random, seed
 from sys import argv, exit
 
@@ -47,7 +46,7 @@ if not pp:
     print 'Using normal k-means'
     initial_means = init
 k = int(argv[1])
-if k < 1 or k > 6:
+if not 1 <= k <= 6:
     print 'K must be a value from 1-6'
     exit()
 
@@ -83,5 +82,4 @@ colors = [[1,0,0], [0,1,0], [0,0,1], [0,0,0], [1,1,1], [.5,.5,.5]]
 for i in range(k):
     c = array(clusters[i])
     scatter(c[:,0], c[:,1], c=colors[i])
-savefig('k-means.pdf')
 show()