[Python/파이썬] sys.stdin.readline()을 이용해 값을 입력 받아 리스트에 넣기

 

sys.stdin.readline

파이썬의 input의 실행 시간이 느려, 이를 해결하기 위해 사용하는 입력 함수이다.

(input과 sys.stdin.readlin의 차이점이 궁금하다면 해당 블로그를 참고하면 도움이 될 것 같습니다.)

 

 

사용법

• 숫자 혹은 문자 하나 입력받는 경우

import sys

N = sys.stdin.sys.stdin.readline()

 

• 띄어쓰기로 구분된 두 개의 문자 혹은 숫자를 변수로 저장하는 경우

import sys

# A,B의 형은 문자
A, B = sys.stdin.readline().split(" ")
 
# 숫자형으로 받고 싶은 경우
A, B = map(int, sys.stdin.readline().split(" "))

 

• 띄어쓰기로 구분된 여러 문자 혹은 숫자를 입력받아 리스트로 저장하는 경우

import sys

test_list = list(sys.stdin.readline().split(" "))

# list의 마지막 원소에 개행 문자(\n)를 제거하고 싶은 경우
test_list = list(sys.stdin.readline().rstrip().split(" "))

# 리스트의 원소들을 숫자로 설정하고 싶은 경우
test_list = list(map(int, sys.stdin.readline().rstrip().split(" ")))

• 엔터로 구분되어 있는 여러 줄 숫자 혹은 문자를 입력받을 경우

import sys

test_list = []

# 입력 받는게 5개인 경우
for i in range(5):
    test_list.append(sys.stdin.readline())
    
 # 개행 문자인 \n를 제거하고 싶은 경우
 for i in range(5):
    test_list.append(sys.stdin.readline().rstrip())
    
 # 숫자로 받고 싶은 경우
 for i in range(5):
    test_list.append(map(int, sys.stdin.readline().rstrip()))