指數沿着N維numpy的陣列的每一維的每個第n個元素

Question

在一個維度也能夠做到：

np.arange(12)[::3] 

取得：

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)