[파이썬] 사용자 입력 공백처리 - input(), sys.stdin.readline(), split(), strip()

파이썬으로 사용자의 입력을 받는 방법 - input(), sys.stdin.readline() 및 입력의 공백 처리하는 방법 - split(), strip()에 대해 알아보자.

 

*이 글은 매번 헷갈려서 구글링하는 블로그 주인을 위해 작성되었습니다. 정확하지 않은 정보가 있다면 댓글로 알려주시면 감사하겠습니다.

 

사용자 입력받기

파이썬에서 사용자의 입력을 받는 방법은 크게 두 가지가 있다.

# 방법1. sys.stdin.readline() 사용
import sys
n1 = sys.stdin.readline()

# 방법2. input() 사용
n2 = input()

두 방식 모두 사용자의 입력을 '문자열'로 받으며, 여기서 기억해야 할 것은 방법1. sys.stdin.readline()이 더 빠르다는 것.

 

사용자 입력 처리하기

사용자의 입력에서 공백을 제거하거나, 입력을 구분해서 각각 다른 변수에 할당하는 경우, strip()과 split()을 주로 사용한다.

(1) strip() :  문자열의 앞뒤 공백이나 특정 문자 삭제하여 문자열로 반환

(2) split() : 문자열 공백이나 특정 문자를 기준으로 구분한 뒤, 리스트로 반환. 언패킹 가능

# 공백제거 방식 2가지 
stri = "ABC  ".strip() # 'ABC'
spli = "ABC  ".split() # ['ABC']

print(len(stri)) # 문자열의 길이, 3
print(len(spli)) # 리스트의 길이, 1

# 언패킹
a, b, c = "A  B  C".split()
d = "A  B  C".split()
print(a) # A
print(b) # B
print(c) # C
print(d) # ['A', 'B', 'C']