반응형
def solution(arr1, arr2):
for i in range(len(arr2)):
for j in range(len(arr2[0])):
arr1[i][j] += arr2[i][j]
return arr1
처음에 풀었을때 위에처럼 풀었음.
def solution(arr1, arr2):
return [[arr1[i][j] + arr2[i][j] for j in range(len(arr2[0]))] for i in range(len(arr2))]
파이썬 답게 한줄로 요약? 했을때
def solution(arr1, arr2):
return [[ v1+v2 for v1,v2 in zip(a1,a2)] for a1,a2 in zip(arr1,arr2)]
다른사람의 풀이에서 내장함수 zip을 이용한걸 보고 이해해볼겸 짜봤다.
반응형
'코딩테스트 연습' 카테고리의 다른 글
비밀지도_파이썬 (0) | 2020.12.24 |
---|---|
x만큼 간격이 있는 n개의 숫자_파이썬 (0) | 2020.12.24 |
프로그래머스_핸드폰 번호 가리기_파이썬 (0) | 2020.12.23 |
프로그래머스_하샤드수_파이썬 (0) | 2020.12.22 |
[카카오 인턴]키패드_누르기_파이썬 (0) | 2020.12.18 |
댓글