not good but great

プログラミング、アート、映画・本の感想について書きます。

numpyで値が大きい上位K件の配列インデックスを取得

gist.github.com

argmaxだと最大値のインデックスしか取れない。上位k件のインデックスが欲しかった。調べて見るとnumpy argpartitionというのがあるらしい。それを使って書いた。