Selection sort in Python – Machine Learning