python - 沿 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)