행렬의 덧셈

Sooin Yoon·2025년 4월 21일

2차원 행렬의 덧셈 -> 2차원 리스트의 각 위치마다 직접 접근해서 더하는 방식

arr1 = [[1,2], [2,3]]
arr2 = [[3,4], [5,6]]

result = []
for i in range(len(arr1)):
	row = []
    for j in range(len(arr2)):
    	sum_value = arr1[i][j] + arr2[i][j]
		row.append(sum_value)
    result.append(row)
result
<<< [[4, 6], [7, 9]]
result = []
for row1, row2 in zip(arr1, arr2):
    row_result = []
    for a, b in zip(row1, row2):
        row_result.append(a+b)
    result.append(row_result)

result

리스트 내포구문 1줄로

result = [[a+b for a, b in zip(row1, row2)] for row1, row2 in zip(arr1, arr2)] 

0개의 댓글