[Python] 2차원 배열 입력받기

안녕하세요! daily_D 입니다! 👩🏻‍💻
오늘은 Python 으로 2차원 배열 입력받는 방법에 대해 공부해봐요!

---

파이썬에서 2차원 배열을 입력받는 방법은 3가지가 있습니다.
아래의 그림과 같이, 가로 A, 세로 B 길이의 배열을 선언한다고 가정해 봅시다!

참고로, 3가지 경우는 모두 배열의 세로(B) 의 크기를 아는 경우에만 사용할 수 있습니다.

 

1. 원소 하나씩 입력받기

arr = [for _ in range(B)] // 2차원 배열의 가로길이 : B

for i in range(B):    
	arr[i] = list(map(int, input().split()))

input, split, map, list 에 대한 이해가 잘 되지 않는다면 [Python] 1차원 배열 입력받기 를 참고해주세요 

 

2. 원소에 list 추가하기

arr = []

for i in range(B):    
	arr.append(list(map(int, input().split())))

 

3. 선언과 동시에 입력받기

arr = [list(map(int, input().split())) for _ in range(B)]