Program to implement selection sort in Python