calculate distortions for k means for elbow method

k_means_distortions.py

#Determine optimal k value for clustering using elbow method
distortions = [] #Initialize array with distortions from each clustering run
K = range(1,11) #Explore k values between 1 and 10
#Run the clustering routine 
for k in K:
    #Convert image into a 1D array
    flat_img = np.reshape(rgb_img,(-1,3))
    kmeanModel = KMeans(n_clusters=k)
    kmeanModel.fit(flat_img)
    distortions.append(kmeanModel.inertia_)