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)