仅当我不知道总维数时如何按一维索引数组？

Question

我有一个（可能是多维的）数组X。它至少有k尺寸。我不知道它有多少个维度。如何k仅按第 维对其进行索引？

例如，如果X是 4 维，k是 3，并且所需的索引是 7，我想得到X[:, :, 7, :]。

Answer 1

使用selectdim：

\n\n

julia> X = reshape(1:2*3*5*7, 5, 3, 7, 2);\n\njulia> selectdim(X, 3, 7)\n5\xc3\x973\xc3\x972 view(reshape(::UnitRange{Int64}, 5, 3, 7, 2), :, :, 7, :) with eltype Int64:\n[:, :, 1] =\n 91   96  101\n 92   97  102\n 93   98  103\n 94   99  104\n 95  100  105\n\n[:, :, 2] =\n 196  201  206\n 197  202  207\n 198  203  208\n 199  204  209\n 200  205  210\n\njulia> selectdim(X, 3, 7) == X[:, :, 7, :]\ntrue\n

\n