在n维数组中边缘化

Question

我正试图找出如何处理多维数组在朱莉娅。我有一个多维数组A = rand(5,5,5)。

我正试图弄清楚如何获得A[1,1,:]或A[1,:,1]或A[:,1,1]与输入m给出的:的位置。

我想出了

indexData = [:,1,2]
indexData[1],indexData[m] = indexData[m],indexData[1]
data = A[indexData[1],indexData[2],indexData[3]]

，但这似乎过于复杂，如果A的维度未知，则无法缩放。有没有更好的解决方法?

Answer 1

以下可能符合要求:

getshaft(A,ii,m) = [A[(i==m?j:ii[i] for i=1:length(ii))...] for j=1:size(A,m)]

考虑下面的例子:

julia> A = reshape(collect(1:27),3,3,3)
3×3×3 Array{Int64,3}:
[:, :, 1] =
 1  4  7
 2  5  8
 3  6  9
[:, :, 2] =
 10  13  16
 11  14  17
 12  15  18
[:, :, 3] =
 19  22  25
 20  23  26
 21  24  27
julia> getshaft(A,(1,2,3),1)
3-element Array{Int64,1}:
 22
 23
 24

第二个参数是元素索引，第三个参数选择维度。getshaft将返回包含第二个参数所选元素的值向量，并沿第三个参数指定的维度返回。第一个参数当然是数组。

—Update—

一个快速的回顾，建议一个更快和更清晰的实现相同的功能:

getshaft(A,ii,m) = A[(i==m?Colon():ii[i] for i=1:length(ii))...]

使用切片索引可能会受益于更快的索引计算或其他后台AbstractArray魔法。