2차원 리스트

Python으로 2차원 리스트 만드는 방법에 대해 살펴보고, 그 외 2차원 리스트 연산을 활용한 예시들을 살펴본다.

---

1. 2차원 리스트 생성하기 

# COLUMN = 가로 길이, ROW = 세로 길이 
array = [[0 for col in range(COLUMN)] for row in range(ROW)]
array = [[0] * COLUMN for _ in range(ROW)] 

# 다음과 같이 작성할 경우, 얕은 복사가 일어나기 때문에 권장하지 x 
array = [[0] * COLUMN] * ROW

2. 2차원 리스트 -> 1차원 리스트로 변환

my_list = [[1, 2], [3, 4], [5, 6]]

# sum 함수 사용 
answer = sum(my_list, [])

# numpy 라이브러리 활용 
import numpy as np

np.array(my_list).flatten().to_list()

# list comprehension 이용 
[element for array in my_list for element in array]