[Python] 2차원 리스트(배열)

 

앞에서 파이썬의 리스트에 관해 배웠는데,
2차원 리스트는 어떻게 만들까?

 

 

 

♣ 2차원 리스트

 

• 직접 값 넣어 2차원 리스트 만들기

age = [[10, 11, 12], [13, 14, 15], [16, 17, 18]]

<결과>
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

 

 

• 크기만 정하고 원하는 값으로 초기화하여 2차원 리스트 만들기

# 행 개수
n = 3

# 열 개수
m = 4

# 0으로 초기화된 2차원 리스트
age = [[0]*m for _ in range(n)]

<결과>
[[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]

 

 

• 원소 값을 직접 입력 받아 2차원 리스트 만들기

# 행과 열을 입력한다.
n = int(input())

age = []

# 입력한 행의 수 만큼 반복한다.
for _ in range(n):
    # 열에 들어갈 원소를 공백으로 구분하여 받는다.
    row = list(map(int, input().split()))
    # row의 값들을 age 리스트에 추가한다.
    age.append(row)

<입력>
3
1 2 3
4 5 6
7 8 9

<결과>
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

 

 

• 컴프리헨션을 사용하여 더 간단하게 2차원 리스트 만들기

# 행을 입력받는다.
n = int(input())

# n만큼 반복하고, 숫자를 입력하여, 열을 채운다.
age = [list(map(int, input().split())) for _ in range(n)]

<입력>
3
1 2 3
4 5 6
7 8 9

<결과>
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

 

 

 

 

♣ 주의

age = [[0]*4]*3

<결과>
[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]

겉으로 보기엔 잘 만들어진 리스트로 보인다. 여기에 age[0][0] = 99를 해보겠다.

 

age = [[0]*4]*3
age[0][0] = 99

<결과>
[[99, 0, 0, 0], [99, 0, 0, 0], [99, 0, 0, 0]]

age[0][0]만 바꿨는데 age[1][0]과 age[2][0]도 값이 바뀌었다. 0번째 행과 1번째 행, 2번째 행의 주소값을 확인해보자.

 

print(age[0]is age[1])
print(age[1]is age[2])

<결과>
True
True

확인 결과 주소값이 같은 것을 알 수 있다. 어떻게 된걸까? 

 

age = [[0]*4]*3

이 코드를 분석해보자. [0]*4는 0을 4번 넣어서 하나의 리스트로 만든다. 그리고 그 만든 리스트를 3번 곱하여 이중 리스트를 만든다. 즉, [0, 0, 0, 0] 리스트를 복사하여 [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]] 만든다는 것이다. 여기서 중요한 점은 복사했다는 점이다. 주소를 복사하여 똑같은 리스트가 된 것이다. 그래서 age[0][0]의 값을 바꾸면 결국 age[1][0]도 같은 주소를 가지고 있어서 함께 바뀌게 된다.