首页 > 解决方案 > 如何在 Julia 的一行中创建两个嵌套的 for 循环

问题描述

我见过几次,有人遇到这样一种情况,他们想在同一行上放置两个 for 循环,并相互嵌套。

只是为了确认一下,这在 Julia 中是否可行,如果可以,它会是什么样子?谢谢!

标签: julia

解决方案


正确,Julia 允许您简洁地表达嵌套for循环

例如,考虑按列顺序填充一个 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

推荐阅读