結果 : if statement in function argument python