[Python] 문자열, 배열 입력 받기

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]]