[Python] input 함수 활용하여 원하는 만큼 리스트 내 원소 자동생성하기

안녕하세요. Harry입니다.

 

본 포스팅의 목적은, 코딩테스트를 준비함에 있어 스스로 공부한 지식을 정리하고자 합니다.

 

코딩테스트를 준비하다보면, 반복문이 1번 돌때마다 나온 값을 한 리스트에 저장해야하는 경우가 있습니다.

 

이런 경우, 보통은 루프 밖에 별도의 리스트를 생성하여 append 함수를 이용하여 리스트를 만들기도 하는데요.

 

이번에는 좀 다르게 활용할 수 있는 방법이 있어 정리하고자 합니다.

 

사용자가 입력하는 숫자만큼 리스트 내 리스트 요소 만들기

사용자의 입력 수의 개수만큼 리스트 내 리스트로 만드는 방법입니다.

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

list(map(int,input().split())) 을 먼저 활용하면, 사용자가 입력하는 문자열을 정수형으로 변경하고, 스페이스바 기준으로 요소로 구분하여 출력합니다.

 

N = int(input())
sample = [list(map(int, input().split())) for i in range(N)]

사용자가 N 이라는 정수형을 입력하면, for 문 내 range 함수를 활용하여 N번 만큼 반복할 수 있도록 설정하구요.

for 문 앞에 list(map(int,input().split())) 을 작성함으로써 for문의 결과를 아래로 들여쓰기하여 표현하지 않고 한줄의 코드로 활용할 수 있도록 작성합니다.

print(sample)

이제 리스트를 출력하면 아래와 같이 나옵니다.

N = 2

입력한 숫자는 2 3 and 5 6 입니다.

결과는 [ [2,3], [5,6]] 으로 나온 것으로 확인할 수 있습니다.

 

감사합니다.