[프로그래머스] [Level 1] 행렬의 덧셈

문제) 행렬의 덧셈

코딩테스트 연습 - 행렬의 덧셈

 

• 내 풀이

def solution(arr1,arr2):
    answer=[]
    for i in range(len(arr1)):
        ans=[arr1[i][j]+arr2[i][j] for j in range(len(arr1[0]))] 
        answer.append(ans)
    return answer

- 생각보다 오래걸림...

- 주의해야할 점

  ans=[arr1[i][j]+arr2[i][j] for j in range(len(arr1[0]))] 

     : 첫 번째 for문이 도는 동안 나오는 모든 값을 하나의 리스트에 함께 넣음

#주의 사항
for i in range(len(arr1)):
	for j in range(len(arr1[0])):
		ans=[arr1[i][j]+arr2[i][j]]
        
#for문 두줄 쓰는 것과 위 코드는 다르다
#결과가 모두 따로따로 리스트로 만들어짐

- 결과

print(solution([[1,2],[3,4]],[[3,4],[5,6]]))

#결과
[[4,6],[8,10]]

 

 

• 다른 사람 풀이 1

def solution(arr1,arr2):
  answer=[[arr1[i][j]+ arr2[i][j] for j in range(len(arr1[0]))] for i in range(len(arr1))]
  return answer

print(solution([[1,2],[3,4]],[[3,4],[5,6]]))

- 내 코드와 유사

- 밖으로 따로 썼던 for문까지 한줄에 합치는 코드

- 하나의 리스트에 for문 하나씩

 

• 다른 사람 풀이 2

def solution(arr1,arr2):
  answer=[[c+d for c,d in zip(a,b)] for a,b in zip(arr1,arr2)]
  return answer

print(solution([[1,2],[3,4]],[[3,4],[5,6]]))

- zip() 함수 사용

 

---

zip()함수 개념

[Python] 내장 함수 zip()