[Python] 2차월 배열 선언 방법

반응형

1차원 배열 선언

rows = 10
graph = [0] * rows

print(graph)

실행결과
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]

 

2차원 배열 선언 ISSUE CASE

2차원 배열 선언에서 1차원 배열에서 선언한 방법과 같이 *을 사용하면 문제가 생깁니다.

rows = 10
cols = 5
graph = [[0] * cols] * rows

print(graph)

실행 결과
[[0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0]]

 

하지만 여기서 값을 넣어보면 문제가 생깁니다.

graph[0][0] = 1

[[1, 0, 0, 0, 0], [1, 0, 0, 0, 0], [1, 0, 0, 0, 0], [1, 0, 0, 0, 0], [1, 0, 0, 0, 0], [1, 0, 0, 0, 0], [1, 0, 0, 0, 0], [1, 0, 0, 0, 0], [1, 0, 0, 0, 0], [1, 0, 0, 0, 0]]

graph의 0행 0열에만 값을 넣었는데, graph의 모든 행의 0열에 값이 변경되었습니다.

 

728x90

Why?

Python에서 *연산자를 이용해 배열을 선언하게 되면, 얕은 복사(shallow copy)가 진행됩니다.

즉, 배열 내의 요소들이 같은 객체를 가리키게 되는 것입니다.

때문에 2차원 배열을 *연산자로 선언했을 때, 값을 넣으면 다른 행에도 영향을 미치는 것입니다.

 

2차원 배열을 어떻게 선언해야 할까?

for문을 이용하여 선언하면 됩니다.

rows = 10
cols = 5
graph = [[0] * cols for _ in range(rows)]

 

Check

graph[0][0] = 1

[[1, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0]]

문제없이 의도한 대로 변경된 것을 확인할 수 있습니다.

 

반응형