提取3d numpy數組的某些列

Question

我想從3d numpy數組中提取某些列。 想像一下3D numpy數組

[[[ 65 234 169]
  [203 191 245]
  [ 36  58 196]
  [207 208 143]
  [251 208 187]]

 [[ 79  69 237]
  [ 13 124  42]
  [104 165  82]
  [170 178 178]
  [ 66  42 210]]

 [[ 40 163 219]
  [142  37 140]
  [ 75 205 143]
  [246  30 221]
  [ 16  98 102]]]

如果-例如-應該提取第2-4列，我想得到一個結果3d數組，例如

[[[203 191 245]
  [ 36  58 196]
  [207 208 143]]

 [[ 13 124  42]
  [104 165  82]
  [170 178 178]]

 [[142  37 140]
  [ 75 205 143]
  [246  30 221]]]

我一直在研究數組索引和切片幾個小時，但是我沒有使它工作。 可以幫助我嗎？

在此先感謝您，並致以最誠摯的問候。

Answer 1

In [12]: array
Out[12]: 
array([[[ 65, 234, 169],
    [203, 191, 245],
    [ 36,  58, 196],
    [207, 208, 143],
    [251, 208, 187]],

   [[ 79,  69, 237],
    [ 13, 124,  42],
    [104, 165,  82],
    [170, 178, 178],
    [ 66,  42, 210]],

   [[ 40, 163, 219],
    [142,  37, 140],
    [ 75, 205, 143],
    [246,  30, 221],
    [ 16,  98, 102]]])

In [13]: array[:, 1:4, :]
Out[13]: 
array([[[203, 191, 245],
    [ 36,  58, 196],
    [207, 208, 143]],

   [[ 13, 124,  42],
    [104, 165,  82],
    [170, 178, 178]],

   [[142,  37, 140],
    [ 75, 205, 143],
    [246,  30, 221]]])