 class="o">&gt;</span> <span class="mi">1000</span><span class="p">:</span>
    <span class="n">cp</span> <span class="o">=</span> <span class="s2">&quot;(compressed to 1000 timesteps)&quot;</span>
    <span class="n">index</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">floor</span><span class="p">(</span><span class="n">np</span><span class="o">.</span><span class="n">linspace</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <