[ Python ] Comprehension으로 다차원 배열 만들기

파이썬에서 리스트 함축(List Comprehension)을 사용하여 다차원 배열을 생성하는 방법은 매우 유용하고 간결합니다.
다음은 다양한 다차원 배열을 만드는 예제들입니다.

1. 2차원 배열 생성

예제 1: 기본 2차원 배열

간단한 2차원 배열을 생성하는 방법입니다.

# 3x3 2차원 배열 생성
array_2d = [[0 for _ in range(3)] for _ in range(3)]
print(array_2d)  # [[0, 0, 0], [0, 0, 0], [0, 0, 0]]

예제 2: 값을 초기화한 2차원 배열

각 요소를 특정 값으로 초기화한 2차원 배열을 생성합니다.

# 3x3 2차원 배열을 각 요소를 i + j로 초기화
array_2d = [[i + j for j in range(3)] for i in range(3)]
print(array_2d)  # [[0, 1, 2], [1, 2, 3], [2, 3, 4]]

2. 3차원 배열 생성

예제 1: 기본 3차원 배열

간단한 3차원 배열을 생성하는 방법입니다.

# 2x2x2 3차원 배열 생성
array_3d = [[[0 for _ in range(2)] for _ in range(2)] for _ in range(2)]
print(array_3d)  # [[[0, 0], [0, 0]], [[0, 0], [0, 0]]]

예제 2: 값을 초기화한 3차원 배열

각 요소를 특정 값으로 초기화한 3차원 배열을 생성합니다.

# 2x2x2 3차원 배열을 각 요소를 i * j * k로 초기화
array_3d = [[[i * j * k for k in range(2)] for j in range(2)] for i in range(2)]
print(array_3d)  # [[[0, 0], [0, 0]], [[0, 0], [0, 1]]]

3. 4차원 배열 생성

4차원 배열도 같은 원리로 생성할 수 있습니다.

예제 1: 기본 4차원 배열

# 2x2x2x2 4차원 배열 생성
array_4d = [[[[0 for _ in range(2)] for _ in range(2)] for _ in range(2)] for _ in range(2)]
print(array_4d)  # [[[[0, 0], [0, 0]], [[0, 0], [0, 0]]], [[[0, 0], [0, 0]], [[0, 0], [0, 0]]]]

예제 2: 값을 초기화한 4차원 배열

# 2x2x2x2 4차원 배열을 각 요소를 i + j + k + l로 초기화
array_4d = [[[[i + j + k + l for l in range(2)] for k in range(2)] for j in range(2)] for i in range(2)]
print(array_4d)  # [[[[0, 1], [1, 2]], [[1, 2], [2, 3]]], [[[1, 2], [2, 3]], [[2, 3], [3, 4]]]]

4. 요약

위의 예제들은 리스트 함축을 사용하여 다차원 배열을 생성하는 방법을 보여줍니다.
리스트 함축은 간결하고 효율적으로 다차원 배열을 초기화할 수 있는 강력한 도구입니다.
다양한 크기와 값을 가지는 배열을 쉽게 만들 수 있으므로, 특정 요구에 맞는 배열을 생성하는 데 매우 유용합니다.