從給定索引堆疊numpy數組的切片

Question

我正在努力對numpy向量執行以下操作。

我想從indices處的vector精加工中獲取previous_n樣本。

就像我要對previous_n樣本進行切片時執行np.take一樣。

例：

import numpy as np

vector = np.array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14])

# number of previous samples
previous_n = 3

indices = np.array([ 5,  7, 12])

結果

array([[ 3,  4,  5],
       [ 5,  6,  7],
       [10, 11, 12]])

Answer 1

好的，這似乎可以滿足我的要求。 在這里找到

def stack_slices(arr, previous_n, indices):
    all_idx = indices[:, None] + np.arange(previous_n) - (previous_n - 1)
    return arr[all_idx]

---

>>> stack_slices(vector, 3, indices)
array([[ 3,  4,  5],
       [ 5,  6,  7],
       [10, 11, 12]])