如何旋转2D整数数组

def rotate(matrix):
    n = len(matrix)
    for layer in range(n // 2):
        first = layer
        last = n - 1 - layer
        for i in range(first, last):
            temp = matrix[first][i]
            matrix[first][i] = matrix[last-i][first]
            matrix[last-i][first] = matrix[last][last-i]
            matrix[last][last-i] = matrix[i][last]
            matrix[i][last] = temp
    return matrix

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

rotated_matrix = rotate(matrix)

for row in rotated_matrix:
    print(row)

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