Как получить индексы максимальных элементов в массиве массивов NumPy?

И, вуаля, вот итоговое решение в одну строку

Как получить индексы максимальных элементов в массиве массивов NumPy?

Так же мне удалось реализовать выдёргивания через комбинацию NumPy функций

Как получить индексы максимальных элементов в массиве массивов NumPy?

Только вашу самописную функцию можно заменить на боолее производительную из numpy

Как видите вывод одинаков
Возьмите себе на заметку вдруг пригодится)
Ещё раз спасибо!

Как получить индексы максимальных элементов в массиве массивов NumPy?

А вот это действительно интересное решение, спасибо большое!

Как получить индексы максимальных элементов в массиве массивов NumPy?

Сергей Соколов, к сожалению нет, здесь нужна чуть более сложная комбинация numpy'ких функций