結果 : if and for statement in one line python