結果 : if statement in one line for loop python