如何從2D python數組中獲取列？

Question

我想知道共同

import numpy as np 
a = np.array([[0,0,0,2,3,0],[3,2,4,0,0,1]]) 

如果我想獲得COL2至COL4，所以結果可能是：

[[0,2,3][4,0,0]] 

我怎麼能做到嗎？ 謝謝。

Answer 1

>>> import numpy as np 
>>> a = np.array([[0,0,0,2,3,0],[3,2,4,0,0,1]]) 
>>> a[:,2:5] 
array([[0, 2, 3], 
     [4, 0, 0]])