如何通过 N 维掩盖“裁剪”多维 numpy 数组？

Question

假设我有一个数组 a，它可以是 N 维的，但为了简单起见，让我们说它：a = np.array([[0,1,2],              [3,4,5]])我也有一个长度为 N 的面具或作物（我不知道正确的术语），所以在这种情况下是这样的：b = [1,2]我如何“裁剪”数组，使其：a = a[:b]这将切片与：a = a[:1,:2]但不管 N 的值如何都会起作用。

Answer 1

使用slice符号 -

a[tuple([slice(None,bi) for bi in b])]

或者用它的简写np.s_——

a[tuple([np.s_[:bi] for bi in b])]