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.