微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决相当于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

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。