如何在Julia的矩阵中找到值的位置?

时间:2016-12-31 07:40:42

标签: arrays matrix julia

我正在使用find这样的3D矩阵A

julia> find(A.==1)
2-element Array{Int64,1}:
1
234
4567

Julia将位置作为索引而不是索引数组。例如,它返回234而不是(1,2,1)。

我看了this question,但我的矩阵非常大,形状为(360,360,360)。我无法使用那里建议的方法。

我尝试研究它的索引模式并使用我编码的函数对其进行转换:

function cmf_p(matrix)
      for a=1:length(matrix);
          aa=matrix[a]
          rd_u_m=ceil(aa/(360^2))
          rd_d_m=floor(aa/(360^2)-1)
          rd_d_t=(aa-rd_d_m*360)/360^2   
          rd_d_p=aa-rd_d_m*360^2-floor(rd_d_t)*360
          println(rd_u_m);
          println(ceil(rd_d_t)*360);
          println(ceil(aa-rd_d_m*360^2-floor(rd_d_t)*360))
      end    
end

但它给了我错误的结果。

如何使用索引并将其转换为我想要的位置?

1 个答案:

答案 0 :(得分:5)

您正在寻找ind2sub

julia> A = eye(3)
3x3 Array{Float64,2}:
 1.0  0.0  0.0
 0.0  1.0  0.0
 0.0  0.0  1.0

julia> inds = find(A.==1.0)
3-element Array{Int64,1}:
 1
 5
 9

julia> [ind2sub(size(A), ind) for ind in inds]
3-element Array{Any,1}:
 (1,1)
 (2,2)
 (3,3)