[Python] 파이썬 - 공백 구분하여 입력받기

참조

• 참조 블로그

소개

• 파이썬으로 코딩테스트 연습을 하던 중, 숫자를 공백으로 구분하여 입력받아야 하는 경우가 있었습니다.
• Python 은 아직 배우고 있는 단계라 해당 문법 정리합니다.

파이썬 공백 구분하여 입력 받기

• 예를 들어, 다음와 같이 Input 값을 입력받아야 하는 경우가 있습니다.

5 8 3
2 4 5 4 6

• 위 경우에 파이썬에서는 다음과 같이 공백을 구분하여 입력받을 수 있습니다.

n, k, m = map(int, input().split())
data = list(map(int, input().split()))

• 위와 같이 선언하면 공백을 구분하여 입력 받을 수 있습니다.

map(int, input().split()) 작동 방식

• input() 으로 받은 문자열을 split() 함수로 공백 단위로 쪼개서 리스트로 나눕니다.
• map 을 이용하여 리스트의 모든 원소에 int() 를 적용합니다.

다른 방법

• 파이썬은 input() 함수로 입력 받다가는 시간이 초과될 수 있다고 합니다.
• 때문에 다음과 같은 방법을 권장합니다.

import sys
arr = sys.stdin.readline().restrip()