NumPy: получить индекс максимального элемента в массиве по одной оси

NumPy: Array Object Exercise-120 с решением

Напишите программу NumPy, чтобы получить индекс максимального элемента в массиве NumPy вдоль одной оси.

Пример решения :

Код Python:

import numpy as np    
a = np.array([[1,2,3],[4,3,1]])
print("Original array:")
print(a)
i,j = np.unravel_index(a.argmax(), a.shape)
print("Index of a maximum element in a numpy array along one axis:")
print(a[i,j])

Пример вывода:

 Исходный массив:
[[1 2 3]
 [4 3 1]]
Индекс максимального элемента в массиве по одной оси:
4

Иллюстрированная презентация:

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Previous: написать программу NumPy для добавления новой строки в пустой массив numpy.
Далее: Напишите программу NumPy, чтобы объединить последовательность массивов вдоль новой оси.