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