CTCI在python中旋转矩阵(2D arrray)

def rotate_matrix(matrix):
    n = len(matrix)
    # 先进行转置操作
    for i in range(n):
        for j in range(i, n):
            matrix[i][j], matrix[j][i] = matrix[j][i], matrix[i][j]
    # 再进行水平翻转操作
    for i in range(n):
        for j in range(n // 2):
            matrix[i][j], matrix[i][n - 1 - j] = matrix[i][n - 1 - j], matrix[i][j]
    return matrix