如何从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]])