[Python] numpy 없이 전치행렬 구현하기(map,zip)

백준 문제를 풀다 보면 종종 리스트 안의 행과 열을 바꿔야 하는 경우가 생긴다.

하지만 코딩테스트에서는 보통 numpy를 사용할 수 없다.

간단한 내장함수를 사용하여 전치행렬을 구현하는 방법을 알아보자.

 

1. zip

for문을 실행할 때 zip 함수로 열의 요소들을 묶어줄 수 있다.

arr = [list(temp) for temp in zip(*arr)]

2. map과 zip

map 함수를 사용하면 for문을 거치는 것보다 더 빠르게 전치행렬을 만들 수 있다.

arr = list(map(list,zip(*arr)))