擅长:python、mysql、java
<pre class="lang-py prettyprint-override"><code>import numpy as np
import itertools
array = np.arange(0,20000)
samplespace = 5000
indices = itertools.chain.from_iterable((ind for ind in range(i, len(array), samplespace) )for i in range(n))
output = array[list(indices)]
</code></pre>
<p>输出:</p>
<pre><code>array([ 0, 5000, 10000, 15000, 1, 5001, 10001, 15001, 2,
5002, 10002, 15002])
</code></pre>