[Python] 한 번에 여러 개 입력(input) 받기

 

Python에서 입력 받는 형식에 대해서 알아봅시다.

1차원, 2차원 리스트 입력도 같이 알아봅시다.

 

 

코드 예시

# 값 두개를 입력 받아서 a와 b에 저장 (띄어쓰기 구분)
a, b = input().split() # 문자열
a, b = map(int, input().split()) # 정수형
a, b = map(float, input().split()) # 실수형

# 1차원 배열 정수형 리스트로 입력 받기
list = list(map(int, input().split()))

# 문자열 여러줄 입력받기
list = [input() for _ in range(n)] # n에 3을 넣으면 3줄 입력 받기

# 문자열 받아서 한글자씩 나누어 더하기

s = input()
for i in range(len(s)):
	sum += int(s[i]) # 문자열 슬라이싱, 문자열 형변환
    
# 띄어쓰기 없이 정수 여러개 입력 받아 2차원 배열로 저장하기
list = [list(map(int, input())) for _ in range(n)]

# 열은 띄어쓰기로 행은 엔터로 구분하여 2차원 배열 저장하기
list = [list(map(int, input().split())) for _ in range(n)]

 

 

또한, 필자의 경우에는 코딩테스트에서 가장 많이 나오는 숫자형을 입력 받고 그 수만큼 리스트를 입력하는 방식을

이렇게 작성합니다.

 

N = int(input())

for i in range(N):
	list = input()