[파이썬] sys.stdin.readline '\n'까지 입력

 

✅ 에러 상황

import sys 
input=sys.stdin.readline

파이썬에서 input()보다 sys.stdin.readline()의 속도가 더 빠르기 때문에,

나는 코드 맨 윗줄에 이 라인들을 추가해주고 입력을 input()으로 그대로 받는 편이다.

 

---

import sys 
input=sys.stdin.readline

data = map(int, input().split())
print(list(data))

#입력: 1 2 3 4 5
#출력: [1, 2, 3, 4, 5]

지금까지는 위와 같이 map을 사용해 공백으로 구분된 문자열을 입력받고 int로 형변환하여 저장했기 때문에

\n까지 입력되어서 문제가 된 적도 없었고,

그래서 당연히 readline으로 입력받으면 \n까지 입력된다는 사실도 모르고 있었다!

 

---

 

import sys 
input=sys.stdin.readline

data = list(input())
print(data)

#입력: 12345
#출력: ['1','2','3','4','5','\n']

그런데 코드를 이렇게 작성하니까 문제가 발생했따

 

 

혹시나 해서 맨 위 두 줄을 지우고 실행했더니,

이번에는 개행 문자가 같이 입력되지 않았다!

 

✅ 해결 과정과 방법

그렇다면 input 대신 sys.stdin.readline을 사용한 것이 문제였다는 말인데,

나는 그동안 input과 readline이 실행속도에만 차이가 있고 동일한 함수로 취급해도 되는 줄 알았는데 아니었나보다...!

 

---

둘의 차이를 검색해보니,

 

input()은 값을 입력받으면 \n 을 삭제시키고 리턴하는 반면,

sys.stdin.readline()은 \n까지 읽어들인다는 차이가 있었다.

 

---

readline() 함수를 사용하여 공백 없는 문자열을 입력받아 리스트에 하나씩 저장하고 싶다면,

.rstrip()을 사용해 마지막 공백을 없애주면 해결이 가능하다.

import sys 
input=sys.stdin.readline

data = list(input().rstrip())
print(data)
#입력: 12345
#출력: ['1','2','3','4','5']