<div dir="ltr"><br><div class="gmail_extra"><div class="gmail_quote">On 21 October 2017 at 21:03, Robert Kern <span dir="ltr"><<a href="mailto:robert.kern@gmail.com" target="_blank">robert.kern@gmail.com</a>></span> wrote:<br><blockquote class="gmail_quote" style="margin:0px 0px 0px 0.8ex;border-left:1px solid rgb(204,204,204);padding-left:1ex"><div dir="ltr"><span class="gmail-"></span>Index with a boolean mask.<div><br></div><div>mask = (tmp_px > 2)</div><div>px = tmp_px[mask]</div><div>py = tmp_py[mask]</div><div># ... etc.<br><br></div></div></blockquote><div><br></div><div>That isn't equivalent, note that j only increases when tmp_px > 2. I think you can do it with something like:</div><div><br></div><div>mask = tmp_px > 2</div><div> j_values = np.cumsum(mask)[mask]</div><div>i_values = np.arange(len(j_values))</div><div><br></div><div>px[i_values] = tmp_i[j_values]<br></div><div><br></div><br></div></div></div>