在Julia中为LU分解构建递归函数

时间:2018-11-05 22:27:31

标签: julia linear-algebra

我对朱莉娅(Julia)非常陌生,并尝试对一些我为递归LU分解编写的代码进行故障诊断。 这是我的全部代码:

`using LinearAlgebra
function recurse_lu(A)
    n=size(A)[1]
    L=zeros(n,n)
    U=zeros(n,n)
    k=n/2
    convert(Int, k)
    A11=A[1:(k),1:(k)]
    A12=A[1:(k),(k+1):n]
    A21=A[(k+1):n,1:(k)]
    A22=A[(k+1):n,(k+1):n]
    if n>2
        L11,U11=recurse_lu(A11)
        L12=zeros(size(A11)[1],size(A11)[1])
        U21=zeros(size(A11)[1],size(A11)[1])
        U12=inv(L11)*A12
        L21=inv(U11)*A21
        L22,U22=recurse_lu(A22-L21*U12)
    else
        L11=1
        L21=A21/A11
        L22=1
        L12=0
        U21=0
        U12=A12
        U22=A22-L21*A12
        U11=A11
    end
    L[1:(k),1:(k)]=L11
    L[1:(k),(k+1):n]=L12
    L[(k)+1:n,1:(k)]=L21
    L[(k)+1:n,(k+1):n]=L22

    U[1:(k),1:(k)]=U11
    U[1:(k),(k+1):n]=U12
    U[(k+1):n,1:(k)]=U21
    U[(k+1):n,(k+1):n]=U22
    return L,U 
end`

这碰到 ArgumentError:无效索引:Float64类型的1.0 ,当我尝试为矩阵计算函数时。我将不胜感激将来的任何提示以及如何解决此问题。我猜我在某种程度上使用了错误的变量类型,但是Julia并没有让您知道问题出在哪里。 非常感谢。

1 个答案:

答案 0 :(得分:3)

该错误是因为您尝试索引具有浮点数的数组,例如:

julia> x = [1, 2, 3]; x[1.0]
ERROR: ArgumentError: invalid index: 1.0 of type Float64

它看起来像来自以下两行:

k=n/2
convert(Int, k)

应该是

k=n/2
k = convert(Int, k)

或者,您可以直接使用整数除法:

k = div(n, 2) # or n ÷ 2

将返回一个Int,您可以使用它建立索引。