結果 : write loop in one line python