arrays - 如何借助 Julia 中的指针访问和更改数组中的元素？

Question

我有一个嵌套数组，想重复获取数组中特定位置的值并更改它。有没有办法用一个更紧凑的指针来做到这一点？我使用 Julia 1.6.0。

我的数组看起来像这样:

Q = Array{Array}(undef, 10,10)
for row in 1:10
    for col in 1:10
        Q[row,col] = Array{Array}(undef, 5,5)
        for i in 1:5
            for j in 1:5
                Q[row,col][i,j] = zeros(9)
            end
        end
    end
end

我目前的解决方案是将变量设置为元素上一级的数组，然后更改它:

tmp = Q[3,3][2,4]
tmp[5] = 1

我想要的是一些直接寻址元素的方法:

tmp = Q[3,3][2,4][5]
tmp = 1

但这并没有改变数组。我阅读了 Ref() 函数的一些内容，但没有完全使用它来获取/更改元素。

提前致谢!

Answer 1

你可以创建一个指针数组:

julia> a = [Ref(0) for i in 1:2, j in 1:2]
2×2 Matrix{Base.RefValue{Int64}}:
 RefValue{Int64}(0)  RefValue{Int64}(0)
 RefValue{Int64}(0)  RefValue{Int64}(0)

现在你可以获得一个元素

b = a[1,2];

你可以改变它:

b[] = 600;

现在我们回到数组看结果

julia> a
2×2 Matrix{Base.RefValue{Int64}}:
 RefValue{Int64}(0)  RefValue{Int64}(600)
 RefValue{Int64}(0)  RefValue{Int64}(0)

您可能还想知道如何获取数据:

julia> getindex.(a)
2×2 Matrix{Int64}:
 0  600
 0    0

不过，对于大多数情况，这不是处理数据的推荐方法。请查看 @view。这是一个等效的例子:

julia> c=zeros(Int,2,2)
2×2 Matrix{Int64}:
 0  0
 0  0

julia> elem = @view c[1,1]
0-dimensional view(::Matrix{Int64}, 1, 1) with eltype Int64:
0

julia> elem .= 999;


julia> c
2×2 Matrix{Int64}:
 999  0
   0  0