Eu tenho um array 2D de (4,5) e outro array 2D de formato (4,2). O segundo array contém os índices inicial e final que eu preciso filtrar do primeiro array, ou seja, eu quero fatiar o primeiro array usando o segundo array.
np.random.seed(0)
a = np.random.randint(0,999,(4,5))
a
array([[684, 559, 629, 192, 835],
[763, 707, 359, 9, 723],
[277, 754, 804, 599, 70],
[472, 600, 396, 314, 705]])
idx = np.array([[2,4],
[0,3],
[2,3],
[1,3]
])
Saída esperada - pode ser qualquer um dos dois formatos a seguir. A única razão para preenchimento com zeros é que arrays 2d de comprimento variável não são suportados.
[[629, 192, 835, 0, 0],
[763, 707, 359, 9, 0],
[804, 599, 0, 0, 0],
[600, 396, 314, 0, 0]
]
[[0, 0, 629, 192, 835],
[763, 707, 359, 9, 0],
[0, 0, 804, 599, 0],
[0, 600, 396, 314, 0]
]