def FdiagPlus1(a,n): f = Fdiag(a,n-1) f = f.col_insert(n-1, zeros(n-1,1)) f = f.row_insert(n-1, Matrix(1,n, lambda i,j: (1 if j==n-1 else 0))) return f