用函数初始化矩阵

时间:2015-08-06 11:25:12

标签: matrix initialization 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

是否有一些快捷方式允许我以更短的方式表达,而不浪费时间分配所有零?

1 个答案:

答案 0 :(得分:8)

直接使用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

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