python 다차원 리스트 생성

개요

계속 까먹는 파이썬 다차원 리스트 생성, 알고리즘 문제 푸는 데 유용한 도구가 되니 더 이상 까먹지 않도록 블로그에 기록합니다.

생성

모든 원소 값 0을 갖는 2차원 리스트를 생성합니다.

n = 2

table = [[0 for _ in range(n)] for _ in range(n)]
# [[0,0], [0,0]]

주의점

아래 프로그램을 실행하여 동일한 결과를 얻을 수 있으나, 일반적인 다차원 리스트를 생성하는 것이 아닌 각 인덱스에 대한 참조 리스트를 생성합니다.

n = 2

table =[([0]*n)] *n
# [[0,0], [0,0]]

table[1][1] 값을 1로 만들고자 할 때, table[0][1] 또한 1로 변경되는 것을 확인할 수 있습니다.

print(table)
# [[0,0], [0,0]]

table[1][1] = 1

print(table)
# [[0,1], [0,1]]

경우에 따라 유용하게 사용할 수 있으나 명확하게 이해한 후 사용해야겠습니다.