[개념] 2차원 배열 입력/ 배열의 합

2차원 배열 입력/ 배열의 합 문제

개념

접근 방법 : 어떻게 입력을 받을 것인지? (2차원 격자형태의 입력을 받는 방법)

• python에서 다음과 같은 2차원 형태를 표현할 수 있는 방법은 리스트이다.

# 입력
1 2 3 4
7 8 9 10
11 12 13 14
15 16 17 18

# 입력받는 형태
arr_2d = [[1, 2, 3, 4], [7, 8, 9, 10], [11, 12, 13, 14], [15, 16, 17, 18]]

• 방법
빈 리스트 하나를 만들고, 한 줄 단위로 입력받는다.

arr_2d = []
for _ in range(4):
	arr_1d = list(map(int, input().split()))
    arr_2d.append(arr_1d)
    
print(arr_2d)

[[1, 2, 3, 4], [7, 8, 9, 10], [11, 12, 13, 14], [15, 16, 17, 18]]

위의 코드에서 arr_2d를 간단하게 한다.

arr_2d = []
for _ in range(4):
    arr_2d.append(list(map(int, input().split())))
    
print(arr_2d)

위의 코드는 list comprehension으로 변경 가능하다.

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

print(arr_2d)

💡 list comprehension 가독성 좋게 작성하는 팁!

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

print(arr_2d)