def solution(arr1, arr2):
total = []
answer = []
for i in range(len(arr1)):
for j in range(len(arr1[i])):
total.append(arr1[i][j]+arr2[i][j])
answer.append(total)
total = []
return answer
def solution2(arr1,arr2):
total=[]
answer=[]
for a,b in zip(arr1,arr2):
for c,d in zip(a,b):
total.append(c+d)
answer.append(total)
total=[]
return answer
zip이라는 개념을 생각하지 못했는데 이 문제를 풀면서 zip에 대해서 한번 더 공부하게 되었다.
def solution3(arr1, arr2):
return [[c + d for c, d in zip(a, b)] for a, b in zip(arr1,arr2)]
이건 위에 zip 코드를 한 줄 코드로 만들었을때의 코드이다
존경스럽다....