input(), sys.stdin.readline 정리

Python에서 입력을 받는 방법은 input()함수와 sys 모듈을 import해 사용하는 방법(sys.stdin.readline())이 있다. 파이썬으로 코딩문제를 풀 때 입력수가 많을 때는 input보다 sys.stdin.readline을 사용하는 것이 시간적으로 유리하다고 다들 알고 있을 것이다. 그래서 나는 왠만한 코딩문제를 풀 때 sys.stdin.readline를 사용한다.

그런데 어느날, 문제를 풀다가 input()과 sys.stdin.readline()의 차이점인 개행 문자 여부에 대해 제대로 인지하지 못한 상태에서 사용을 하다가 해맨적이 있었다. 그래서 이번기회에 정리를 해보고자 한다.

---

input()

• prompt 인자가 있으면, 개행없이 표준 출력으로 기록된다.
• input() 으로 부터 받은 입력의 개행문자를 벗겨서(stripping) 문자열로 변환한 값을 리턴한다.
• EOF(End of file)이 읽혀지면, EOFError를 발생시킨다. 

EOF(End of file)이란?

파일의 끝을 의미하며 데이터 소스로부터 더 이상 읽을 수 있는 데이터가 없음을 나타낸다. C언어에서는 EOF는 -1이라는 상수값을 가지고 있다.

 

즉, input은 입력받은 값의 개행(줄바꿈) 문자를 삭제시켜서 받아온다는 것이다. 기존의 문자열에서 개행 문자를 삭제시키는 작업을 취하기 때문에 속도가 느려지는 것이다.

---

sys.stdin.readline

sys 모듈은 Python 인터프리터가 제공하는 변수와 함수를 직접 제어할 수 있게 해주는 모듈이다. 

stdin 은 sys 모듈안에 있는 파일 객체이고 공식문서에 따르면,

• stdin 은 파이썬 인터프리터가 표준 입력에 사용하는 파일 객체
• readline() 은 파일 객체의 메소드 중 하나로 read(), readlines() 와 같이 파일 객체를 읽을 때 사용한다.

sys.stdin.readline() 은 sys 라는 모듈의 파일 객체 stdin 의 메소드 중 readline() 을 사용한다는 의미이다. readline() 은 입력을 읽을 때 한 번에 한 줄씩 읽는데, 이 말은 여러 줄의 입력이 있을 때 한 줄을 읽고 나면 그 다음 줄을 가리킨다는 뜻이다.

 

즉, sys.stdin.realine은 개행(줄바꿈) 문자를 포함한 값을 받아온다는 것이다. 따라서 input보다 속도면에서 빠르다 볼 수 있다.

 

정리하자면, 입력수가 많거나 속도가 중요한 문제인 알고리즘 문제의 경우에는 sys.stdin.readline을 쓰는것이 유리하다. 하지만 문자열을 출력하는 문제에서는 input을 사용하는 것이 실수를 줄일 수 있다.