相当于2D列表理解的“地图”

问题描述

在1d中，我可以使用以下任意一个：

[i for i in 1:5] 

或

map(1:5) do i 
    i
end

都生产

[1,2,3,4,5]

是否可以在更高维度上使用map？例如复制

[x + y for x in 1:5,y in 10:13] 

产生

5×4 Array{Int64,2}:
 11  12  13  14
 12  13  14  15
 13  14  15  16
 14  15  16  17
 15  16  17  18

解决方法

您可以这样做：

julia> map(Iterators.product(1:3,10:15)) do (x,y)
         x+y
       end
3×6 Array{Int64,2}:
 11  12  13  14  15  16
 12  13  14  15  16  17
 13  14  15  16  17  18

您写的理解是collect(x+y for (x,y) in Iterators.product(1:5,10:13)) ，。注意括号(x,y)，因为do函数得到一个元组。与x,y不同，它有两个参数：

julia> map(1:3,11:13) do x,y
         x+y
       end
3-element Array{Int64,1}:
 12
 14
 16

,

这当然不是您要寻找的map等价物，但是在这种情况下，您可以使用带有矢量和转置矢量的广播：

x = 1:5
y = (10:13)'
x .+ y

在REPL：

julia> (1:5) .+ (10:13)'
5×4 Array{Int64,2}:
 11  12  13  14
 12  13  14  15
 13  14  15  16
 14  15  16  17
 15  16  17  18