我试图弄清楚如何在julia中处理多维数组。我有一个多维数组A = rand(5,5,5)

我试图找出如何通过输入m给出的A[1,1,:]的位置来获取A[1,:,1]A[:,1,1]:

我想出了

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

但这似乎过于复杂,并且如果A的尺寸未知,则无法缩放。有解决这个问题的更好方法吗?

最佳答案

以下可能符合要求:

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将沿着由第三个参数指定的维度返回值向量,包括第二个参数选择的元素。当然,第一个参数是数组。

- - 更新 - -

快速浏览后,建议您更快,更干净地实现相同功能:
getshaft(A,ii,m) = A[(i==m?Colon():ii[i] for i=1:length(ii))...]

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

10-06 14:52