Elbow Method Python