是否有一种简单的方法可以一次对数组中的几个元素进行采样？

Question

为了更快地进行一阶分析，我需要对一组数据进行采样。如果我想一次检查一个点，我可以通过切片来完成：

import numpy as np
array = np.arange(0,20000)
samplespace = 5000
sampledarray = array[::samplespace]

但是，我需要分析几个元素。我能想到的唯一方法是使用for循环：

import numpy as np
array = np.arange(0,20000)
samplespace = 5000
n = 3
sampledarray = array[0::samplespace]
for i in range(1,n):
    arraysample_i = array[i::samplespace]
    indices = np.linspace(i,len(sampledarray),len(arraysample_i)).astype(int)
    sampledarray = np.insert(sampledarray,indices,arraysample_i)
print(sampledarray)
>>> [ 0 1 2 5000 5001 5002 10000 10001 10002 15000 15001 15002 ]

我担心如果我使用大数组并在多个维度中进行采样，那么循环运行会花费很多时间。有没有更简单、更快的方法

Answer 1

import numpy as np
array = np.arange(0,20000)
samplespace = 5000
n = 3
sampledarray = np.stack([array[i::samplespace] for i in range(n)]).flatten(order='F')

输出：

array([    0,     1,     2,  5000,  5001,  5002, 10000, 10001, 10002,
       15000, 15001, 15002])