結果 : if and for loop in single line python