二维数组(矩阵)层旋转代码。尝试优化代码以进行矩阵层旋转

def rotate_matrix(matrix):
    if not matrix or not matrix[0]:
        return matrix

    rows, cols = len(matrix), len(matrix[0])
    start_row, start_col = 0, 0
    end_row, end_col = rows - 1, cols - 1

    while start_row < end_row and start_col < end_col:
        prev = matrix[start_row + 1][start_col]

        # 上边
        for i in range(start_col, end_col + 1):
            curr = matrix[start_row][i]
            matrix[start_row][i] = prev
            prev = curr

        start_row += 1

        # 右边
        for i in range(start_row, end_row + 1):
            curr = matrix[i][end_col]
            matrix[i][end_col] = prev
            prev = curr

        end_col -= 1

        # 下边
        for i in range(end_col, start_col - 1, -1):
            curr = matrix[end_row][i]
            matrix[end_row][i] = prev
            prev = curr

        end_row -= 1

        # 左边
        for i in range(end_row, start_row - 1, -1):
            curr = matrix[i][start_col]
            matrix[i][start_col] = prev
            prev = curr

        start_col += 1

    return matrix