[ 파이썬(Python) ] 2차원 행렬에서 세로(column) 열만 추출하기

📍 2차원 행렬에서 세로(column) 열만 추출하기

1부터 25까지 순서대로 있는 5 * 5행렬에서 세로 (열(column))만 추출하는 방법을 알아보자. 총 3가지 방법으로 2중 반복문, lambda, zip함수를 사용했다. (더 좋은 방법이 있으면 댓글로 알려주세요!!) 결과는 모두 동일하다.

 

# 1 ~ 25까지 배열 만들기
arr = [[0] * 5 for _ in range(5)]
cnt = 0

for i in range(5):
    for j in range(5):
        cnt += 1
        arr[i][j] = cnt

# 2중 반복문
result = [[arr[j][i] for j in range(5)] for i in range(5)]

# lambda 함수
result = [list(map(lambda x: x[i], arr)) for i in range(5)]

# zip 함수
result = list(map(list, zip(*arr)))

👉🏽 [[1, 6, 11, 16, 21], [2, 7, 12, 17, 22], [3, 8, 13, 18, 23], [4, 9, 14, 19, 24], [5, 10, 15, 20, 25]]