<div dir="ltr"><div><div>In what way does it not work? Does it error out at the `arr = arr[mask]` step? Or is it that something unexpected happens?<br><br></div>I am guessing that you are trying to mutate the px, py, pz, w, x, y, z arrays? If so, that for-loop won't do it. In python, a plain simple assignment merely makes the variable point to a different object. It doesn't mutate the object itself.<br><br></div>Cheers!<br>Ben Root<br><br></div><div class="gmail_extra"><br><div class="gmail_quote">On Fri, Oct 27, 2017 at 11:43 AM, Andrei Berceanu <span dir="ltr"><<a href="mailto:berceanu@runbox.com" target="_blank">berceanu@runbox.com</a>></span> wrote:<br><blockquote class="gmail_quote" style="margin:0 0 0 .8ex;border-left:1px #ccc solid;padding-left:1ex">Hmm, so how come this doesn't work now?<br>
<br>
mask = ((px > 2.) & ((py**2 + pz**2) / px**2 < 1.))<br>
<br>
for arr in (px, py, pz, w, x, y, z):<br>
    arr = arr[mask]<br>
<div class="HOEnZb"><div class="h5"><br>
On Mon, 23 Oct 2017 15:05:26 +0200 (CEST), "Andrei Berceanu" <<a href="mailto:berceanu@runbox.com">berceanu@runbox.com</a>> wrote:<br>
<br>
> Thank you so much, the solution was much simpler than I expected!<br>
><br>
> On Sat, 21 Oct 2017 23:04:43 +0200, Daπid <<a href="mailto:davidmenhur@gmail.com">davidmenhur@gmail.com</a>> wrote:<br>
><br>
> > On 21 October 2017 at 22:32, Eric Wieser <<a href="mailto:wieser.eric%2Bnumpy@gmail.com">wieser.eric+numpy@gmail.com</a>><br>
> > wrote:<br>
> ><br>
> > > David, that doesn’t work, because np.cumsum(mask)[mask] is always equal<br>
> > > to np.arange(mask.sum()) + 1. Robert’s answer is correct.<br>
> > ><br>
> > Of course, you are right. It makes sense in my head now.<br>
> > ______________________________<wbr>_________________<br>
> > NumPy-Discussion mailing list<br>
> > <a href="mailto:NumPy-Discussion@python.org">NumPy-Discussion@python.org</a><br>
> > <a href="https://mail.python.org/mailman/listinfo/numpy-discussion" rel="noreferrer" target="_blank">https://mail.python.org/<wbr>mailman/listinfo/numpy-<wbr>discussion</a><br>
><br>
><br>
> ______________________________<wbr>_________________<br>
> NumPy-Discussion mailing list<br>
> <a href="mailto:NumPy-Discussion@python.org">NumPy-Discussion@python.org</a><br>
> <a href="https://mail.python.org/mailman/listinfo/numpy-discussion" rel="noreferrer" target="_blank">https://mail.python.org/<wbr>mailman/listinfo/numpy-<wbr>discussion</a><br>
<br>
<br>
______________________________<wbr>_________________<br>
NumPy-Discussion mailing list<br>
<a href="mailto:NumPy-Discussion@python.org">NumPy-Discussion@python.org</a><br>
<a href="https://mail.python.org/mailman/listinfo/numpy-discussion" rel="noreferrer" target="_blank">https://mail.python.org/<wbr>mailman/listinfo/numpy-<wbr>discussion</a><br>
</div></div></blockquote></div><br></div>