如何在Julia中的一行中创建两个嵌套的for循环

时间:2019-12-29 05:03:16

标签: julia

我见过几次,有人要把两个for循环放在彼此嵌套的同一行上。

只需确认一下,Julia是否有可能,如果是这样,它看起来像什么?谢谢!

1 个答案:

答案 0 :(得分:10)

正确,Julia可让您简洁地表达嵌套的for loops

例如,考虑按列顺序填充3x3矩阵:

julia> xs = zeros(3,3)
3×3 Array{Float64,2}:
 0.0  0.0  0.0
 0.0  0.0  0.0
 0.0  0.0  0.0

julia> let a = 1
           for j in 1:3, i in 1:3
               xs[i,j] = a
               a += 1
           end
       end

julia> xs
3×3 Array{Float64,2}:
 1.0  4.0  7.0
 2.0  5.0  8.0
 3.0  6.0  9.0

上面的循环等效于这个更详细的版本:

julia> let a = 1
           for j in 1:3
               for i in 1:3
                   xs[i,j] = a
                   a += 1
               end
           end
       end

甚至更高维度都支持此语法(!):

julia> for k in 1:3, j in 1:3, i in 1:3
           @show (i, j, k)
       end
相关问题