Python Sort Lambda If Function