Как замаскировать «обрезать» многомерный массив Numpy по N измерениям?

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])]