Python Apply If Elif