Function Julia映射将二维数组返回到三维数组的函数

我有一个返回二维数组的函数。 如果我

map

将此函数映射到

UnitRange

我将得到一个二维数组的一维数组。 是否有一种简洁的方法将其收集到三维阵列中

y = rand(3,3)
f(x) = y * x
map(f, 1:9)

这将导致

9元素数组{Array{Float64}，2}

而不是

Array{Float64,3}

。

---

使用理解功能也可以达到同样的效果。

您可以使用

cat

功能，如下所示：

cat（dims，A…）

沿指定的方向连接输入数组 可编辑dims中的尺寸

例如，下面的con

cat

启用了三维中的两个二维阵列

julia> A = collect(reshape(1:9, 3, 3));

julia> B = collect(reshape(10:18, 3, 3));

julia> cat(3, A, B)
3×3×2 Array{Int64,3}:
[:, :, 1] =
 1  4  7
 2  5  8
 3  6  9

[:, :, 2] =
 10  13  16
 11  14  17
 12  15  18

将此应用于您的情况，您可以跨阵列阵列

foldl

foldl((A,B) -> cat(3, A, B), map(f, 1:9))

不过，对于更干净的解决方案，您可以使用splat（

…

）操作符将所有数组结果传递给

cat

函数的单个调用

cat(3, map(f, 1:9)...)

---

或者，让函数以

UnitRange

作为输入并始终返回三维数组是更好的解决方案吗？

cat（3，map（f，1:9）…

是一种可能性

[i*j代表y中的i，j=1:9]

是另一个没有定义

f

的数组，它确实产生了一个三维数组，但结果仍然是一个数组数组

9x1x1x1数组{array{Float64,2}，3}

而不是

3x3x9数组{Float64,3}

实际上，cat表达式确实有效，我错过了省略号

…

。现在，它的工作如预期。