[python] sys.stdin.readline()

SeongMok Hong·2022년 9월 3일
0
post-custom-banner

sys.stdin.readline()

파이썬에서 보통 입력을 받을때는 Input함수를 사용하였는데, sys 모듈을 import 한 후 sys.stdin.readline() 함수를 사용하면 빠르게 입력 받을 수 있다.
백준 15552번 문제 참고

import sys

# n = input()
n = sys.stdin.readline()
print(n)

sys.stdin.readline() 함수 사용 시 개행문자를 같이 입력받기 때문에, 개행문자를 제외하고 입력을 받기 위해서는 뒤에 .rstrip() 을 추가해주면 된다.

import sys

n = sys.stdin.readline().rstrip()
print(n)

좀더 편하게 사용위하여 input 에 할당하여 사용할 수 있다.

import sys

input = sys.stdin.readline
a = input()
b = input().rstrip()

print(a)
print(b)

위에서 언급했듯이, 개행문자를 같이 입력받기 때문에, 아래와 같이 익명 함수 lambda 를 사용하여 개행문자 제거까지 한번에 할당할 수 있다.

import sys
input = lambda: sys.stdin.readline().rstrip()

...
profile
안녕하세요. 홍성목입니다.
post-custom-banner

0개의 댓글