Как транслировать разные фрагменты для каждого ряда?

Question

Я работаю на массивном двумерном массиве и пытаюсь найти способ доступа к различным фрагментам в строке.Предпочтительно эксплуатировать нудистское вещание таким образом, чтобы я передавал массив фрагментов следующим образом:

A = np.array([[1,2,3,4,5],[1,4,9,16,25],[1,8,27,64,125]])
A[[2:,3:,4:]]
>>> array([[3,4,5],[16,25],[125]])

Я знаю, что я написал выше, является поддельным, но вы понимаете, в чем суть.Я хотел бы передать массив t = [2,3,4], чтобы каждый член этого массива (обозначается как t_i) приводил к t_i: (этот член и двоеточие после).

Спасибо

Answer 1

То, что вы хотите, возможно, не возможно с вещанием.Но понимание списка может работать:

np.array([a[i+2:] for i,a in enumerate(A)])
#array([array([3, 4, 5]), array([16, 25]), array([125])], dtype=object)