IT 세계의 후아

[python]2차원 리스트 생성 및 접근, 값 추가 본문

Coding/Python

[python]2차원 리스트 생성 및 접근, 값 추가

후__아 2024. 6. 13. 14:25

※ 2차원 리스트(배열)

접근할 때 배열 틀 제대로 갖추지 않고 무턱대고 값을 저장하면 index error 남!!

#프로그래머스 lv1 행렬의 덧셈
def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)):
        answer.append([])	# arr1, arr2와 같이 2차원 배열로 틀 갖추기
        for j in range(len(arr1[i])):
            answer[i].append(arr1[i][j] + arr2[i][j])
    return answer
    
    # 다른 사람 풀이
    return [list(map(sum, zip(*x))) for x in zip(A, B)]
    return [[c + d for c, d in zip(a,b)] for a, b in zip(A,B)]

 

answer = [[]]

...

answer[i][j] = arr1[i][j] + arr2[i][j]

>> IndexError

answer는 아직 2x2 배열 상태가 아님!

틀을 똑같이 만들어준 후에 값을 추가하는 형식으로~

 

cf) https://dojang.io/mod/page/view.php?id=2293