インデックス一次元でN次元numpyのアレイ

Question

の各次元に沿ってすべてのn番目の要素である

array([0, 3, 6, 9]) 

効率的であり、どのようなN次元のために同じことをする方法？例えば

、所与：

np.random.random((10, 15, 20)) 

各次元に沿った内のすべての第三の要素を取得する方法？

Answer 1

3Dのために、あなたが行うことができます：

a = np.random.random((10, 15, 20)) 

a[::3,::3,::3].shape 
# (4, 5, 7) 

a.shape 
# (10, 15, 20) 

を任意の大きさのために、あなたはslice使用する可能性があります：

a[[slice(None,None,3)] * a.ndim].shape 
# (4, 5, 7)