lass="o">*</span><span class="n">np</span><span class="o">.</span><span class="n">pi</span>
    <span class="n">nval</span> <span class="o">=</span> <span class="mi">51</span><span class="o">*</span><span class="mi">24</span>
    <span class="n">omega</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">([</span><span class="mf">1.</span><span class="p">,</span> <span class="mf">2.</span><span class="p">,</span> <span class="mf">3.</spa