結果 : write for loop in one line python