我试图弄清楚如何在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魔术。