2차원 배열 회전

아현·2021년 8월 27일
0

Algorithm Note

목록 보기
16/18

1. 반복문 이용


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

def rotate_matrix90(matrix):
	col = len(matrix[0])
    new_matrix = [[0] * len(matrix) for _ in range(len(matrix))]
    for i in range(len(matrix)):
        for j in range(len(matrix[0])):
            new_matrix[i][j] = matrix[col-j-1][i]
    return new_matrix


print(rotate_matrix90(matrix))

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



2. zip 이용


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

def rotate_matrix90(matrix):
    return list(zip(*matrix))
    
print(rotate_matrix90(matrix[::-1]))

#[(7, 4, 1), (8, 5, 2), (9, 6, 3)]
profile
For the sake of someone who studies computer science

0개의 댓글