考虑下面的一维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_
,它允许进行多个非连续选择,但是它需要提供手动的开始和结束索引位置,并且所有数组都组合在一起。我需要上述所需格式的输出。