結果 : if else in one line python with for loop