我在朱莉娅中寻找的东西就像一个理解,但是对于矩阵而不是向量。如果我有一些单变量函数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
是否有一些快捷方式允许我以更短的方式表达,而不浪费时间分配所有零?
答案 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
通过在末尾添加变量范围,这适用于任意数量的维度。