Not sure if I'm doing something wrong, but to my surprise I was unable to import the argpartition module from numpy. I'm kind of suprised that Quantopian doesn't have this as it would be a huge time saver in comparison with using argsort, since argsort sorts the entire array where as argpartition only sorts the needed n length of an array.