파이썬 한 번에 입력 여러개 받는 방법(map함수)

파이썬에서 값을 입력받을 때는 input() 함수를 주로 사용합니다.

 

입력이 간단하다는 장점이 있지만 입력된 값이 전부 문자열로 저장된다는 단점이 존재합니다.

 

따라서 input() 함수는 입력값을 문자열로 저장하기 때문에 띄어쓰기도 함께 문자열로 저장됩니다.

 

한 번에 여러개의 입력을 받기 위해 어떤 방법을 사용해야 하는지 알아보겠습니다.

 

 

 

값 하나 입력받기

 

값이 하나이고 입력값의 형식이 정해져 있는 경우 int(), float() 함수들로 형변환을 하여 변수에 저장할 수 있습니다.

 

str_a = input()
int_a = int(input())
float_a = float(input())

 

 

 

값 여러개 입력받기

 

한 줄에 여러개의 값을 입력받아 저장해야하는 경우가 있습니다.

 

이런 상황에서는 split() 함수를 사용합니다.

 

# 띄어쓰기를 기준으로 나누기
a, b = input().split()

# 쉼표를 기준으로 나누기
a, b = input().split(',')

# 콜론을 기준으로 나누기
a, b = input().split(':')

 

 

split() 함수의 괄호 안에 아무것도 넣지 않으면 띄어쓰기를 기준으로 문자를 나누어 저장합니다.

 

괄호 안에 문자열을 넣으면 해당 문자열을 기준으로 앞, 뒤 값을 나누어 저장합니다.

 

 

 

값 여러개를 형태를 변환하여 입력받기

 

만약 입력값이 문자열이 아닌 정수나 실수 등으로 형변환을 시키고 싶다면 map() 함수를 사용합니다.

 

a, b = map(함수, input().split())

# int형으로 저장
a, b = map(int, input().split())

# float형으로 저장
a, b = map(float, input().split())

 

 

map(함수, 데이터) 형식으로 사용할 수 있습니다.

 

각각의 데이터에 함수를 적용하여 나온 리턴값을 저장합니다.

 

위 예제는 입력값이 두 개일 경우를 가정한 상황입니다.

 

더 많은 입력값이 존재할 경우 리스트 형식으로 저장할 수 있습니다.

 

 

arr = list(map(int, input().split()))

# 입력값: 1 2 3 4 5
# arr: [1, 2, 3, 4, 5]

 

list() 함수를 이용하여 입력값을 리스트 형식으로 저장할 수도 있습니다.