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