[Python] 한 번에 연속으로 입력 받기 (split, map)

 

백준 문제를 풀다보면 위와 같이 한 번에 연속으로 입력 받아야 하는 경우가 많다. 하지만 계속 잊어서 글로 남겨둔다.

 

 

Python에서 입력을 받을 때는  input 함수 를 사용한다.

input은 값을 문자열로 저장하기 때문에, 정수나 실수 형태로 저장하려면 int(input()) or float(input()) 형태로 저장할 수 있다.

 

여러 개의 값을 입력 받으려면  split() 을 사용한다.

split도 마찬가지로 값을 문자열로 저장하기 때문에  map 함수 에 int or float을 앞에 지정해주어 정수나 실수로 변환한다.

 

 

# map 함수

: 리스트의 요소를 지정된 함수로 처리하는 함수. 

  보통 여러 개의 데이터를 한 번에 다른 형태로 바꾸기 위해 사용한다.

 

  ≫ 형태

 map ( function, iterable ) 

    - 첫번째 매개변수는 적용시킬 함수, 두 번째 매개변수는 반복가능한 자료형 (리스트, 튜플 등)

    - map 함수의 반환 값은 map 객체라서 list나 tuple로 형 변환을 시켜주어야 한다.

    - 함수는 int와 타입, 람다함수, 일반 함수 모두 가능

 

 

 

#1. 값 두 개를 입력받아 변수 A, B에 저장

A, B = input().split()	#문자열로 입력받음

A, B = map(int, input().split())	#정수형으로 입력받음
A, B = map(float, input().split())	#실수형으로 입력받음

 

#2. 리스트로 입력받아 저장

L = list(map(int, input().split()))		#입력: 1 2 3 / 출력: [1, 2, 3]