arr = np.arange(100 * 100 * 100).reshape(100, 100, 100)
np.random.shuffle(arr)
indices = np.argpartition(arr.flatten(), -2)[-2:]
np.vstack(np.unravel_index(indices, arr.shape)).T
arr[97][99][98]
arr[97][99][99]
Получить индексы N самых высоких значений в ndarray