結果 : single line for loop python with if else