python - 从 Julia 中另一个数组的特定列中获取子数组

Question

我的目标是使用 Julia 从矩阵中提取某些列，然后将这些列变成一个新数组。例如，在 Python 中，如果我定义

x=np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]])
y=[1,3]
x[:,y]

这给了我想要的结果:

array([[ 2,  4],
       [6,  8],
       [10, 12],
       [14, 16]])

但是，在 Julia 中，如果我定义

x=[1 2 3 4; 5 6 7 8; 9 10 11 12; 13 14 15 16]
y=[1 3]
view(x, :, [1 2])

它返回一个 4×1×2 数组，这对我正在做的事情来说是不可取的:

  4×1×2 view(::Array{Int64,2}, :, [1 2]) with eltype Int64:
  [:, :, 1] =
   1
   5
   9
   13

   [:, :, 2] =
   2
   6
   10
   14

如何编写上面的代码，使我的输出为 4x2 数组； IE。，

   1   2
   5   6
   9   10
   13  14

Answer 1

您已经写了 [1 2]这是一个 1 行矩阵，应该写成 [1, 2]这是一个向量:

julia> x=[1 2 3 4; 5 6 7 8; 9 10 11 12; 13 14 15 16]
4×4 Array{Int64,2}:
  1   2   3   4
  5   6   7   8
  9  10  11  12
 13  14  15  16

julia> view(x, :, [1, 2])
4×2 view(::Array{Int64,2}, :, [1, 2]) with eltype Int64:
  1   2
  5   6
  9  10
 13  14