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

gist.github.com argmaxだと最大値のインデックスしか取れない。上位k件のインデックスが欲しかった。調べて見るとnumpy argpartitionというのがあるらしい。それを使って書いた。 MNISTで nearest neighbor / int か float か,それが問題です / np.argpartition - まんぼう日記 numpy.argpartition — NumPy v1.12 Manu…