Using Lambda To Sort Array In Python