julia - 如何在 Julia 的一行中创建两个嵌套的 for 循环
问题描述
我见过几次,有人遇到这样一种情况,他们想在同一行上放置两个 for 循环,并相互嵌套。
只是为了确认一下,这在 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