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