Python 배열 입력받기

이도윤·2022년 11월 19일
0

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

profile
Java 백엔드 개발자

0개의 댓글