在julia中执行索引方式矩阵运算

时间:2017-07-03 15:57:26

标签: matrix multidimensional-array julia

我想对矩阵执行索引式操作。我知道您可以编写常规函数并在矩阵的每个条目上执行它,例如

function foo(x::Int64)
    return x * 2
end
myArray = [1 2 3; 4 5 6]
foo.(myArray)

我如何做x * x.elementCol + x.elementrow这样的事情?基本上是以下代码并行:

function goo(x::Array{Int64,2})
    for j = 1:size(x,2)  
        for i = 1:size(x,1)
            x[i,j] = (x[i,j] * j) + i    
        end
    end
    return x
end

1 个答案:

答案 0 :(得分:4)

你可以写:

x .= x .* indices(x, 2)' .+ indices(x, 1)