파이썬 - 행렬 덧셈

코딩테스트 연습 - 행렬의 덧셈 | 프로그래머스 스쿨 (programmers.co.kr)

def solution(x,y):
    for k in range(len(x)) :
        for t in range((len(x[0]))) :
            x[k][t]+=y[k][t]        
    return x

def sumMatrix(A,B):
    answer = [[A[i][j] + B[i][j] for j in range(len(A[0]))] for i in range(len(A))]
    return answer

이건 상상이 가능한데

 

 

 

def sumMatrix(A,B):
    answer = [[c + d for c, d in zip(a,b)] for a, b in zip(A,B)]
    return answer

def sumMatrix(A,B):
    return [list(map(sum, zip(*x))) for x in zip(A, B)]

zip을 많이 안써서 익숙하지가 않다.

여러개의 순회 가능한(iterable) 객체를 인자로 받고, 각 객체가 담고 있는 원소를

튜플 형태로 차례로 접근할 수 있는 반복자(iterator)를 반환한다.

 numbers = [1, 2, 3]
 letters = ["A", "B", "C"]
 for pair in zip(numbers, letters):
 #for i in range(3):
 	#pair = (numbers[i], letters[i])
     print(pair)

#(1, 'A')
#(2, 'B')
#(3, 'C')

 

 

 

 

import numpy as np

def sumMatrix(A,B):
    A_np = np.array(A)
    B_np = np.array(B)
    result = A_np + B_np
    return result.tolist()

+ numpy