"p">((</span><span class="n">z_comb</span> <span class="o">==</span> <span class="n">array</span><span class="p">[</span><span class="n">z_indices</span><span class="p">,</span> <span class="n">sample_index</span><span class="p">])</span><span class="o">.</span><span class="n">all</span><span class="p">(</span><span class="n">axis</span><span class="o">=</span><span class="mi">1</span><span class="p">))[</span><span class="mi">0</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span