본문 바로가기
Python/Study

파이썬 입력 받기 sys.stdin.readline()

by notcherry 2023. 5. 27.

 

sys.stdin.readline()

->아래와 같이 여러 줄을 입력 받아야 할 때 사용!

출처 취업코딩테스트

 

예제 1. 처음 입력받은 수 만큼, 두 수를 입력 받고 합을 출력하는 문제

import sys

for i in range(int(input())):
        a,b = map(int, sys.stdin.readline().split())
        print(a+b)

map()은 반복가능한 객체에 대해 각각 요소를 지정된 함수로 처리해주는 함수이다.

a,b는 숫자를 입력 받으면 map함수에 의해 int형으로 변환하여 split()함수에 의해 나눠서 저장된다.

split은 스페이스 텝 엔터 등에 의한 공백을 기준으로 문자열을 나눈다.

 

 

만약 list 형식으로 저장하고 싶다면?

import sys

numList = list(map(int, sys.stdin.readline().split()))

 

세로로 저장하고 싶다면?

import sys

numArr = []
for i in range(int(input())) :
	numArr.append(list(map(int, sys.stdin.readline().split())))

 

 

sys.stdin.readline() 사용하려면 import sys 해줘야하고 list에서 쌓는 것은 append!