[python] map과 split을 이용하여 입력값을 변수에 할당하기

Surf in Data·2022년 4월 13일

목록 보기

4/15

python으로 코딩테스트를 준비하다보면 항상 입력값이 주어지고 그에 대한 변수를 만들어 입력값을 변수에 할당하는 것 부터 시작된다.

먼저 map과 split에 대해 알아보겠다.

첫번째 인자는 함수
두번째 인자는 반복가능한 자료형
map 함수의 반환 값은 map 객체이기 때문에 사용하기 위해서는 list 혹은 tuple로 만들어줘야한다.

test_list = ["1", "2", "3", "4"]
test = list(map(int, test_list))
print(test)

[1, 2, 3, 4]

split함수는 문자열을 나눠서 리스트로 만들어준다.
split()의 default 는 띄어쓰기이다.

word = "hi hi yi yi hi yi"
print(word.split())
print(word.split(" ", 2))

['hi', 'hi', 'yi', 'yi', 'hi', 'yi']
['hi', 'hi', 'yi yi hi yi']

ex)
정수 3개가 공백을 두고 입력된다면 다음과 같이 정수3개를 변수에 할당할 수 있다.

num = list(map(int,input().split())) #10 100 1000 입력
print(num)

[10, 100, 1000]

문자열을 할당시켜야할경우는 split만으로 할당가능하다
ex)
공백이 하나인 문자열 입력

word1, word2 = input().split() # hi yi 입력
print(word)
print(num)

hi
yi

study blog

python