如何从numpy数组中提取所需的维数

Question

鉴于我有一个三维[3,500,500]的numpy数组，我怎样才能提取一维为[1,500,500]？

import numpy as np 
my_array = np.ones((3,500,500),dtype=int,order='C') 
print (my_array) 

Answer 1

那些？

>>> my_array[0,:,:] 
array([[1, 1, 1, ..., 1, 1, 1], 
     [1, 1, 1, ..., 1, 1, 1], 
     [1, 1, 1, ..., 1, 1, 1], 
     ..., 
     [1, 1, 1, ..., 1, 1, 1], 
     [1, 1, 1, ..., 1, 1, 1], 
     [1, 1, 1, ..., 1, 1, 1]]) 
>>> my_array[0,:,:].shape 
(500, 500)