Algorithm/문제풀이 Python

[꿀팁] 2차원 배열(N*N) 행,열 쉽게 뒤집기

nauni 2021. 8. 21. 13:06

zip 기본 내장함수와 *list (unpacking) 을 사용하여, 2중 for 문을 사용하지 않고 row, column 을 뒤집을 수 있다.

 

N*N 배열이 아닌 각 배열 크기가 다른 2차원 배열인 경우 예상하지 않은 결과가 나올 수 있다.

# 2차원 배열 행, 열 뒤집기

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

flip_arrays = list(map(list, zip(*arrays)))

#1
print(arrays)
#2
print(flip_arrays)
#3
for item in zip(*arrays):
    print(item)
# 출력

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