Python 2차원배열 전치

1. zip(*graph)

• *은 언패킹(unpacking) 연산자이다. 이를 통해 graph의 각 행이 개별 인수로 zip 함수에 전달된다.
예를 들어, graph가 아래와 같다면:

PythonCopy

graph = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

zip(*graph)는 아래와 같은 식으로 동작한다.

PythonCopy

zip([1, 2, 3], [4, 5, 6], [7, 8, 9])

• zip 함수는 전달받은 인수들의 같은 인덱스에 있는 요소를 묶어 튜플로 반환한다.

PythonCopy

zip([1, 2, 3], [4, 5, 6], [7, 8, 9])
# 결과: [(1, 4, 7), (2, 5, 8), (3, 6, 9)]

2. map(list, zip(*graph))

• zip이 반환하는 것은 튜플이므로, 이를 리스트로 변환하는 작업이 필요하다.

• map(list, zip(*graph))는 zip이 반환한 각 튜플을 리스트로 변환하는 역할을 한다.

3. list(...)

• map 객체는 반복 가능한 객체이지만, 이를 리스트로 변환하려면 list() 함수가 필요하다.

• list(map(list, zip(*graph)))는 이 결과를 다시 리스트로 감싸서 최종적으로 전치된 2차원 리스트를 반환한다.

요약

예시

원래의 graph

graph = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

전치된 graph

graph = list(map(list, zip(*graph)))
# 결과:
[
    [1, 4, 7],
    [2, 5, 8],
    [3, 6, 9]
]