1차원 배열 입력받기
map함수를 이용하면 for문을 쓰지 않고 한줄에 여러개의 정수를 입력받을 수 있다.
입력값을 공백을 기준으로 나눈다.
num_list = list(map(int, input().split()))
입력
1 2 3 4 5
출력
[1, 2, 3, 4, 5]
한줄에 정수형 변수 여러개 입력받기
이 경우에는 받을 변수의 개수를 지정할 수 있다. 만약 입력받은 개수가 변수 개수와 다르면 에러 발생
a, b, c, d = map(int, input().split())
입력
1, 2, 3, 4
출력
>> print(a)
1
>> print(b)
2
한줄에 문자열 변수 여러개 입력받기
a, b = input().split()
입력
123 321
출력
>> print(a, b)
123 321
>> print(type(a))
<class 'str'>
문자열 여러줄 입력받기 - v.1
graph = [input() for _ in range(a)]
입력
WLLWWWL
LLLWLLL
LWLWLWW
LWLWLLL
WLLWLWW
출력
>> print(graph)
['WLLWWWL', 'LLLWLLL', 'LWLWLWW', 'LWLWLLL', 'WLLWLWW']
문자열 여러줄 입력받기 - v.2
이번에는 한줄에 띄어쓰기 없이 정수를 여러개 입력받았을 때, 2차원 배열 형태로 저장하는 방법이다.
maze = [list(map(int, input())) for _ in range(n)]
입력
101111
101010
101011
111011
출력
[[1, 0, 1, 1, 1, 1], [1, 0, 1, 0, 1, 0], [1, 0, 1, 0, 1, 1], [1, 1, 1, 0, 1, 1]]
2차원 배열 입력받기
arr = [list(map(int, input().split())) for _ in range(n)]
입력
1 3 3 6 7
8 13 9 12 8
4 16 11 12 6
2 4 1 23 2
9 13 4 7 3
출력
>> print(arr)
[[1, 3, 3, 6, 7], [8, 13, 9, 12, 8], [4, 16, 11, 12, 6], [2, 4, 1, 23, 2], [9, 13, 4, 7, 3]]