如何從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)