指数沿着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)