<div dir="ltr"><div class="gmail_extra"><div class="gmail_quote"><br></div><div class="gmail_quote">On Thu, Oct 1, 2015 at 11:46 AM, Alex Rogozhnikov <span dir="ltr"><<a href="mailto:alex.rogozhnikov@yandex.ru" target="_blank">alex.rogozhnikov@yandex.ru</a>></span> wrote:<br><blockquote class="gmail_quote" style="margin:0px 0px 0px 0.8ex;border-left-width:1px;border-left-color:rgb(204,204,204);border-left-style:solid;padding-left:1ex">Hi, I have written some numpy tips and tricks I am using, which may be interesting to you.<br>
This is quite long reading, so I've splitted it into two parts:<br>
<br>
<a href="http://arogozhnikov.github.io/2015/09/29/NumpyTipsAndTricks1.html" rel="noreferrer" target="_blank">http://arogozhnikov.github.io/2015/09/29/NumpyTipsAndTricks1.html</a></blockquote><div><br></div><div>The recommendation of inverting a permutation by argsort'ing it, while it works, is suboptimal, as it takes O(n log(n)) time, and you can do it in linear time:<div><br></div><div><div><font face="monospace, monospace">In [14]: import numpy as np</font></div><div><font face="monospace, monospace"><br></font></div><div><font face="monospace, monospace">In [15]: arr = np.random.rand(10)</font></div><div><font face="monospace, monospace"><br></font></div><div><font face="monospace, monospace">In [16]: perm = arr.argsort()</font></div><div><font face="monospace, monospace"><br></font></div><div><font face="monospace, monospace">In [17]: perm</font></div><div><font face="monospace, monospace">Out[17]: array([5, 0, 9, 4, 2, 8, 6, 7, 1, 3])</font></div><div><font face="monospace, monospace"><br></font></div><div><font face="monospace, monospace">In [18]: inv_perm = np.empty_like(perm)</font></div><div><font face="monospace, monospace"><br></font></div><div><font face="monospace, monospace">In [19]: inv_perm[perm] = np.arange(len(perm))</font></div><div><font face="monospace, monospace"><br></font></div><div><font face="monospace, monospace">In [20]: np.all(inv_perm == perm.argsort())</font></div><div><font face="monospace, monospace">Out[20]: True</font></div></div><div class="gmail_extra"><br><div class="gmail_quote">It does require two lines of code, so for small stuff it is probably good enough to argsort, but it gave e.g. np.unique a nice boost on larger arrays when we applied it there.</div><div class="gmail_quote"><br></div><div class="gmail_quote">Jaime</div><div> <br></div></div></div><blockquote class="gmail_quote" style="margin:0px 0px 0px 0.8ex;border-left-width:1px;border-left-color:rgb(204,204,204);border-left-style:solid;padding-left:1ex"><br>
<a href="http://arogozhnikov.github.io/2015/09/30/NumpyTipsAndTricks2.html" rel="noreferrer" target="_blank">http://arogozhnikov.github.io/2015/09/30/NumpyTipsAndTricks2.html</a><br>
<br>
Comments are welcome, specially if you know any other ways to make this code faster (or better).<br>
<br>
Regards,<br>
Alex.<br>
<br>
_______________________________________________<br>
NumPy-Discussion mailing list<br>
<a href="mailto:NumPy-Discussion@scipy.org" target="_blank">NumPy-Discussion@scipy.org</a><br>
<a href="https://mail.scipy.org/mailman/listinfo/numpy-discussion" rel="noreferrer" target="_blank">https://mail.scipy.org/mailman/listinfo/numpy-discussion</a><br>
</blockquote></div><br><br clear="all"><div><br></div>-- <br><div class="gmail_signature">(\__/)<br>( O.o)<br>( > <) Este es Conejo. Copia a Conejo en tu firma y ayúdale en sus planes de dominación mundial.</div>
</div></div>