"Пример numpy.argmax.\nСреда: встроенный MicroPython FEMM 5.\nПлоский индекс наибольшего элемента.\nРезультат: int для argmin/argmax; вещественное число для std; числовой скаляр для остальных операций без оси."

import numpy as np
a=np.array([1+2j,2+0j,3+1j])
y=np.argmax(a)
assert abs(y-(2))<1e-8
print(y)