[Python] 입력 처리 정리 | input | readline | 계속 입력 받기 | 리스트로 입력받기

Python을 처음에 사용할 때 혹은 백준을 풀기 위해서

입력값을 받아야 할 때가 있습니다

 

오늘은 입력을 받기 위한 여러가지 방법과

이후 처리 방법들을

정리해보도록하겠습니다

 

 

~목차~

1. input()

2. readline()

3. 계속 입력 받기

    3-1) for + stdin
    3-2) while + try + readline

4. 여러값이 한 줄에 있을 때

   : 리스트로 입력받기

 

 

 

---

 

1. input()

Python에서 입력값을 받을 수 있는 가장 간단한 방법입니다.

따로 추가할 모듈 없이 바로 입력을 받을 수 있습니다!

 

간단한 코드를 같이 봅시다.

number = input()
print("숫자 %s를 입력하셨습니다!" % number)

#24
#숫자 24를 입력하셨습니다!

여기서 주의 할 점은 input으로 입력받은 값은 str 타입입니다!

(print문에 '%d' 변환문자를 사용하면 오류 발생)

 

변수를 출력하는 방법은 첨부한 글을 참고하시기 바랍니다

 

 

또한 input에는 입력 받기 전 문자열을 출력할 수 있는 기능이 있습니다

age = input("나이를 입력해주세요 : ")
print("10년뒤에 %d살이 되시는 군요!" % (int(age) + 10))

#나이를 입력해주세요 : 24
#10년뒤에 34살이 되시는 군요!

input의 괄호안에 문자열을 입력해주면 됩니다!

입력 받은 값은 str이므로 int형으로 변환하여 계산하고 출력해줄 수 있습니다

 

 

 

2. readline()

input과 비교하여 빠른 속도를 자랑하는 readline입니다!

백준을 풀다보면 시간초과를 만날 때가 많습니다.

이를 방지하기 위해서, 많은 입력을 받을 일이 있다면 readline사용을 권장드립니다

 

기본적으로 readline을 사용하기 위해서는 sys모듈을 import해주어야합니다

import sys
input = sys.stdin.readline

name = input()
print("Hello, ",  name)

#kim sinyoung
#Hello,  kim sinyoung

위 코드처럼 input에 readline을 덮어씌워서 평소처럼 input을 사용하지만 속도는 훨씬 향상 되었습니다!

 

주의 할 점은 '\n', 즉 개행문자까지 포함하여 입력받습니다

import sys
input = sys.stdin.readline

name = input()
age = input()
print("Hello, ",  name, "your are ", age, "years old")

#ksy
#24
#Hello,  ksy
# your are  24
# years old

Hello, ksy your are 24 years old

해당 결과값을 기대했는데 뜻대로 출력 되지 않은 것을 확인할 수 있습니다.

 

하지만 문자열은 인자로 넘겨줄 수 없습니다..ㅠ

입력 전 문자열을 출력하고 싶다면 print문을 써야겠죠?!

 

 

3. 계속 입력 받기

3-1) 입력의 끝을 알 수 없을 때

3-2) 계속 입력 받고싶을 때

 

3-1) For + stdin

import sys

for line in sys.stdin:
	print(line)

for문 안에 조건문을 작성하여 반복을 종료할 수 있습니다.

 

3-2) while + try + readline

import sys
input = sys.stdin.readline

while True:
	try:
    	number = input()
        print(number)
    except EOFError:
    	break

 

while문을 통해 무한 반복하지만 EOF를 만나면 except에 걸리게 되면서 반복문을 나올 수 있습니다.

 

 

4. 여러값이 한 줄에 있을 때

: 리스트로 입력받기

한 줄에 여러 값이 존재하고 이를 변수 또는 리스트에 저장하고 싶으시군요!

 

여러값 변수에 입력받기

import sys
input = sys.stdin.readline

N, M = input().split()

print(int(N) + 3, int(M) - 1)

#2 6
#5 5

split 함수를 이용해서 각 변수에 입력 받을 수 있습니다

 

input은 항상 str타입이기 때문에 int를 붙여서 값을 조정해야된다는 불편이 있습니다.

값을 받을 때 원하는 데이터 타입으로 받을 수 있는 map 함수가 있습니다!

import sys
input = sys.stdin.readline

N, M = map(int, input().split())

print(N + M)

#2 3
#5

 

 

여러값 리스트에 입력받기

변수가 아닌 리스트로 저장하고 싶다면!

map 함수를 list 함수로 감싸주면 됩니다

import sys
input = sys.stdin.readline

alist = list(map(int, input().split()))

print(alist)

#1 2 3 4 5
#[1, 2, 3, 4, 5]

 

만약 여러 줄을 한꺼번에 리스트로 넣고싶다면!

import sys
input = sys.stdin.readline

n = int(input())
alist = []
for _ in range(n):
	alist.append(list(map(int, input().split())))
    
print(alist)

#3
#1 2 3 4
#5 6 7 8
#9 0
#[[1, 2, 3, 4], [5, 6, 7, 8], [9, 0]]