numpy получить индекс из n самых больших значений
numbers = np.array([1, 3, 2, 4])
n = 2
indices = (-numbers).argsort()[:n]
Courageous Cow
numbers = np.array([1, 3, 2, 4])
n = 2
indices = (-numbers).argsort()[:n]