帮我写一个二维高度矩阵旋转的python代码

当你需要将一个二维矩阵顺时针或逆时针旋转90度时，可以使用Python的简单数组操作来实现。这里是一个经典的二维矩阵顺时针旋转的实现：

def rotate(matrix):
    # 获取矩阵的行数和列数
    n = len(matrix)
    
    # 创建一个新的转置矩阵用于存储旋转后的结果
    rotated_matrix = [[0] * n for _ in range(n)]
    
    # 遍历原矩阵，将每个元素放入新矩阵对应的位置
    for i in range(n):
        for j in range(n):
            rotated_matrix[j][n - 1] = matrix[i][j]
            
    return rotated_matrix

# 示例使用
matrix_to_rotate = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

rotated_matrix = rotate(matrix_to_rotate)
print("旋转后的矩阵：")
for row in rotated_matrix:
    print(row)