Python - 다차원 배열 초기화

 

1차원 배열 초기화

>>> [0]*2
[0, 0]

 

초기화 후 값 수정

>>> a = [0] * 2
>>> a
[0, 0]
>>> a[0] =1
>>> a
[1, 0]

 

for문을 사용한 방법

>>> [i*10 for i in range(10)]
[0, 10, 20, 30, 40, 50, 60, 70, 80, 90]

 

2차원 배열 초기화

>>> [[0]*2]*3
[[0, 0], [0, 0], [0, 0]]

 

2차원 배열에 의도와 다른 현상

>>> a = [[1]*2]*3
>>> a
[[1, 1], [1, 1], [1, 1]]
>>> a[0][0] = 2
>>> a
[[2, 1], [2, 1], [2, 1]]

 

2차원 배열에 개선

>>> m = [[i*j for i in range(10)] for j in range(3)]
>>> m
[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
 [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]]
>>> m[0][0] = 1
>>> m
[[1, 0, 0, 0, 0, 0, 0, 0, 0, 0],
 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
 [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]]

 

다차원 배열 생성

>>> m = [[[i*j*k for i in range(2)] for j in range(3)] for k in range(4)]
>>> m
[[[0, 0], [0, 0], [0, 0]], [[0, 0], [0, 1], [0, 2]], [[0, 0], [0, 2], [0, 4]], [
[0, 0], [0, 3], [0, 6]]]

 

다차원 배열 생성 순서

>>> [[int(j) for j in input().split()] for i in range(int(input()))]
2
3 4 5
6 7
[[3, 4, 5], [6, 7]]