結果 : if else in one line for loop python