matrix - 用函数初始化矩阵

Question

我在 Julia 中寻找类似推导式的东西，但寻找的是矩阵而不是向量。如果我有一些单变量函数 f(x)我想要一个充满 f(i) 的数组为 i在 1..10 ， 我可以做这个:

[f(i) for i = 1:10]

如果我有一些二变量函数 g(i,j)我想要一个来自 i=[1,10]; j=[1,10] 的矩阵充满了我可以这样做的功能:

M = zeros (10,10)

for i in 1:10
     for j in 1:10
         M[i,j] = g(i,j)
     end
end

是否有一些捷径可以让我以更短的方式表达它，而不会浪费时间分配所有零？

Answer 1

只需使用 multidimensional comprehension直接地:

julia> g(x,y) = 2x+y
g (generic function with 1 method)

julia> [g(i,j) for i=1:10, j=1:10]
10x10 Array{Int64,2}:
  3   4   5   6   7   8   9  10  11  12
  5   6   7   8   9  10  11  12  13  14
  7   8   9  10  11  12  13  14  15  16
  9  10  11  12  13  14  15  16  17  18
 11  12  13  14  15  16  17  18  19  20
 13  14  15  16  17  18  19  20  21  22
 15  16  17  18  19  20  21  22  23  24
 17  18  19  20  21  22  23  24  25  26
 19  20  21  22  23  24  25  26  27  28
 21  22  23  24  25  26  27  28  29  30

通过在末尾添加变量范围，这适用于任意数量的维度。