如何快速提取二维数组的不连续数据

代码

import numpy as np 
a=np.array([[1,2,3,4,5],[2,3,4,5,6],[12,23,34,56,54]])
print(a.take([1,3,4],1))
print(a.take([1,3,4],1).take([0,1],0))

结果

[[ 2  4  5]
 [ 3  5  6]
 [23 56 54]]
[[2 4 5]
 [3 5 6]]

归纳

a.take([1,3,4],1):抽取维度1的第1,3,4位置的元素，通常1表示列。
take([0,1],0)：抽取维度0的0,1位置的元素，0表示行。