从numpy数组中的多个索引位置选择固定范围的元素

时间:2019-06-11 06:56:02

标签: python arrays numpy numpy-broadcasting

考虑下面的一维np.array

a =  np.array([20, 70, 68, 36, 86, 12, 89, 32,  3, 52, 94,  6, 26, 95, 16, 82, 42,60,  5, 94])
我想从多个索引位置开始,从上述Universe中选择固定长度的子数组。例如,假设以下索引位置

mask_idx = [5,7,11]

和从起始索引位置开始固定5个元素的长度

我希望输出是这样的

out = [[12, 89, 32,  3, 52],
       [32,  3, 52, 94,  6],
       [ 6, 26, 95, 16, 82]]

我遇到过np.r_,它允许进行多个非连续选择,但是它需要提供手动的开始和结束索引位置,并且所有数组都组合在一起。我需要上述所需格式的输出。

0 个答案:

没有答案