Если ваш вопрос заключается в том, как использовать функции map () и np.argmax () вместе, то вот пример, на который вы можете сослаться, где каждая строка передается функции np.argmax () с помощью map (),Надеюсь, что это очистит ваш вопрос.
a = np.array ([[2,1,4,5], [7,5,9,8]])
print (list (карта (np.argmax, a)))
вывод: [3, 2]