Как функция numpy.argmax () может обрабатывать список с цифрами и символами

Question

Как определить строку с наибольшим значением в index = 0, даже если index = 1 является символом?

У меня есть массив списков с двумя позициями, где первая позиция - это число ивторая позиция - символ.

A=np.array([[150,"A"],[9,1],[9,14]])
B=np.argmax(A, axis=0)

Я должен получить это:

[0 2]  

Но я получаю это:

[1 0]

Я не уверен, чтовывод для второй позиции - но для первой позиции он должен быть 0. Что я ошибся в функции?Даже нарезка не сработала.

Answer 1

Я нашел решение.Я не знаю, столкнусь ли я позже с проблемами!

A=np.array([(150,"A"),(30,"B"),(9,"14")], dtype='object')
c=np.argmax(A, axis=0)