[Python] 입력 받기 : sys.stdin.readline()과 input()

sys.stdin.readline() 과 input()

 

파이썬에서 보통 input()으로 값을 입력받지만, 반복문으로 여러 줄을 입력받아야 할 때는 시간 초과 문제가 날 수 있다고 한다. 이럴 경우 import sys로 모듈을 불러오고  sys.stdin.readline() 을 사용하는 것이 좋다.

---

😀 sys 모듈은 무엇일까?

 

sys 모듈은 파이썬 인터프리터와 관련된 정보와 기능을 제공하는 모듈이다.sys 모듈은 강제로 스크립트를 종료할 수 있고, 파이썬 모듈들이 저장된 위치도 알 수 있는 등 시스템을 제어할 수 있는 모듈로 생각된다.

 

➕ stdin 이란?

stdin 은 모든 대화식 입력에 사용된다.우리가 흔히 키보드로 입력하는 행위 뿐 아니라, 파일 등의 넓은 범위의 입력을 의미한다.input() 호출을 포함하고 있다는 점에 주목.

 

📍 input() 대신 sys.stdin.readline()을 사용하는 이유

 

1. input() 이 호출되면 Prompt 문자열을 화면에 출력하고 사용자 입력을 기다린다.

2. 이때 표준 입력은 키보드.

3. 사용자가 키를 하나씩 누르면 이에 대응하는 데이터가 버퍼에 들어간다.

4. Enter 키를 누르면 개행문제(ex. \n)가 입력되며, 버퍼의 입력이 종료된 것으로 간주된다.

5. 입력된 문자열은 해당 시스템의 콘솔 입출력 인코딩을 사용하여 디코드되어 유니코드 문자열로 변환된다.

6. input() 함수는 변환된 문자열 값을 반환하면서 종료한다. 

 

즉, 이러한 단계를 거치기 때문에 input() 은 비교적 속도가 느린 편이다.

 

 

 

📍 sys.stdin.readline()

 

✔ sys.stdin.readline() 은 문자열로 입력을 받는다.

따라서 만약 입력받은 문자열을 정수, 실수, 리스트로 사용할 때는 적절하게 함수를 사용하여 처리해줘야 한다.

 

✔ 개행 문자 "/n"을 같이 입력받는다.

예를 들어, "Hello, World!" 라는 문자열을 입력받았을 때.

sys.stdin.readline() 의 입력으로는 "Hello, World!/n" 이 들어오게 된다.

 

import sys

temp = sys.stdin.readline()
print(temp)  # "Hello, World!\n"

 

\n 이라는 개행 문자가 같이 들어오기 때문에 따로 문자열 처리를 해주어야 한다.

 

 

 

📍 한 줄에 하나의 값 입력받기

 

1️⃣ 하나의 정수 입력받기

예시 코드

import sys

num = int(sys.stdin.readline())  # 5입력
print(num)

출력 결과

5

• 하나의 정수 값을 입력받을 때 사용할 수 있다.
• 문자열로 입력받기 때문에 int() 를 사용해서 정수로 변경해야 한다.

 

2️⃣ 개행 문자와 함께 문자/문자열 입력받기

 

예시 코드

import sys

s = sys.stdin.readline()  # Hello, World! 입력
print(s)

출력 결과

Hello, World!

• 개행 문자와 함께 문자열을 입력받을 수 있다.

 

 

3️⃣ 개행 문자와 없이 문자/문자열 입력받기

 

예시 코드

import sys

""" 입력
Hello, World!
Hello, World!
Hello, World!
"""

s1 = sys.stdin.readline().strip()
s2 = sys.stdin.readline().strip('\n')
s3 = sys.stdin.readline().strip('\n')

print(s1, s2, s3)

출력 결과

Hello, World! Hello, World! Hello, World!

• strip() 을 이용하여 개행 문자 없이 문자열을 입력받을 수 있다.
• strip() 은 문자열 양 옆의 \n, \t 등을 제거할 수 있다.
• \t는 공백에 해당하기 때문에 인자없이 strip()을 쓰면 공백도 같이 제거된다.
• 확실하게 strip('\n') 을 쓰거나, rstrip('\n') 등 인자를 쓰는 걸 추천한다.
• rstrip() 은 오른쪽의 공백이나 개행 문자를 삭제한다.
• strip(), rstrip() 의 매개변수로 특정 문자를 넣으면 해당 문자를 문자열에서 제거한다.

 

➕ 개행 문자 확인 방법

예시 코드

import sys

""" 입력
Hello, World!
Hello, World!
"""

arr = []
s = sys.stdin.readline()
arr.append(s)
print(s)  # 문자열 출력
print(arr)  # 문자열 넣은 배열 출력

출력 결과

Hello, World!

['Hello, World!\n']

• 한 칸이 띄워진 것을 확인할 수 있다.
• 배열에 넣어보면 "\n"이 들어있는 걸 확인할 수 있다.

 

 

📍 한 줄에 2개 이상의 값 입력받기

 

1️⃣ 두 개 이상의 정수 입력받기

예시 코드

import sys

N, M = map(int, sys.stdin.readline().split())	#5 4 입력
print(N, M)

출력 결과

5 4

• map() 을 이용하여 두 개 이상의 정수를 입력받을 수 있다.
• 리스트로 반환하고 싶다면 map()을 list()로 감싸줘야 함.

 

➕ 만약 map() 함수가 없다면?

예시 코드

a, b = int(sys.stdin.readline().split())
print(a,b)

출력 결과

 

• 타입 에러 발생한다.
• sys.stdin.readline().split()은 리스트로 반환되는데,
• int() 함수 안의 매개변수에는 리스트가 들어오면 안되고, 문자열/객체/숫자 타입이 들어와야 하기 때문이다.

 

1️⃣ 두 개 이상의 문자/문자열 입력받기

예시코드

import sys

""" 입력
he l wo d
"""

strings = sys.stdin.readline().split()
print(strings)

출력 결과

['he', 'l', 'wo', 'd']

• 공백을 기준으로 문자/문자열을 입력받을 수 있다.
• sys.stdin.readline().split() 는 리스트를 반환한다.

 

📍 여러 줄에 걸쳐서 입력받기

 

1️⃣ 두 개 이상의 정수 입력받기

입력 예시

4 3
1 0 0
0 1 0
0 0 1
1 1 1

• 여러 줄에 걸쳐 공백을 기준으로 떨어져 있는 정수를 입력받을 때 사용할 수 있다.

 

(1) 2차원 배열로 입력받기

 

예시 코드

import sys

rows, cols = map(int, sys.readline().split())
matrix = []

for r in range(rows):
	matrix.append(list(map(int, sys.stdin.readline().split())))
    
print(matrix)

출력 결과

[[1, 0, 0], [0, 1, 0], [0, 0, 1], [1, 1, 1]]

 

(2) 2차원 배열로 입력받기 - 한 줄에

 

예시코드

import sys

rows, cols = map(int, sys.stdin.readline().split())
matrix = [list(map(int, sys.stdin.readline().split())) for r in range(rows)]

print(matrix)

출력 결과

[[1, 0, 0], [0, 1, 0], [0, 0, 1], [1, 1, 1]]

• 위와 동일하지만 한 줄에 입력받을 수 있는 코드

 

 

📍 공백 없이 주어지는 정수 입력받기

 

입력 예시

4 3
100
010
001
111

• 여러 줄에 걸쳐 공백 없이 주어지는 정수를 입력받을 때 사용할 수 있다.

 

1) 1차원 배열로 입력받기

 

예시 코드

import sys

rows, cols = map(int, sys.stdin.readlint().split())
maxtrix = [[0]] * rows	# 우선적으로 초기화

for r in range(rows):
	matrix[r] = sys.stdin.readline().strip()
    
print(matrix)

출력 결과

['100', '010', '001', '111']

 

 

2) 1차원 배열로 입력받기 - 한 줄에

 

예시 코드

import sys

rows, cols = map(int, sys.stdin.readline().split())
matrix = [sys.stdin.readline().split() for r in range(rows)]

print(matrix)

출력 결과

['100', '010', '001', '111']