python - 扩展切片 numpy 数组的步骤

Question

我的问题与此类似:subsampling every nth entry in a numpy array

假设我有一个如下所示的数组:a = [1,2,2,2,3,4,1,2,2,2,3,4,1,2,2,2,3,4....]

如何扩展切片，以便在特定间隔内对三个元素进行切片？ IE。我怎样才能从数组中切出 2s？我相信基本切片在这种情况下不起作用。

Answer 1

您可以通过单独的索引来做到这一点。

我们想从索引为 1 的元素开始，取 3 个元素，然后跳过 3 个元素:

a = np.array([1, 2, 2, 2, 3, 4, 1, 2, 2, 2, 3, 4, 1, 2, 2, 2, 3, 4])

start = 1
take = 3
skip = 3

indices = np.concatenate([np.arange(i, i + take) for i in range(start, len(a), take + skip)])

print(indices)
print(a[indices])

输出:

[ 1  2  3  7  8  9 13 14 15]
[2 2 2 2 2 2 2 2 2]