Python 数据结构小技巧

 1.numpy选取数据的列

ta = np.arange(12).reshape([3,4])

out(array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]]))

ta[:,2]

array([ 2,  6, 10])

对于X[:,0];
是取二维数组中第一维的所有数据

对于X[:,1]
是取二维数组中第二维的所有数据

对于X[:,m:n]
是取二维数组中第m维到第n-1维的所有数据

对于X[:,:,0]
是取三维矩阵中第一维的所有数据

对于X[:,:,1]
是取三维矩阵中第二维的所有数据

对于X[:,:,m:n]
是取三维矩阵中第m维到第n-1维的所有数据