NumPy найти максимальное значение в строке и вернуться к индексу столбца

Question

У меня есть массив из 180295 * 10, называемый lda_trans, строка означает слова, а столбец означает 10 тем.

array([[0.01841009, 0.01840699, 0.35798764, ..., 0.38443892, 0.01841072,
        0.12870054],
       [0.1       , 0.1       , 0.1       , ..., 0.1       , 0.1       ,
        0.1       ],
       [0.1       , 0.1       , 0.1       , ..., 0.1       , 0.1       ,
        0.1       ],
       ...,
       [0.0416964 , 0.62473603, 0.0416964 , ..., 0.04169395, 0.04169796,
        0.04169232],
       [0.03772096, 0.03775132, 0.66048403, ..., 0.03771698, 0.03772411,
        0.0377139 ],
       [0.03754747, 0.03756587, 0.66206395, ..., 0.03754399, 0.037551  ,
        0.03753927]])

Теперь я хочу вернуться к названию столбца максимального значения каждой строки, я знаю только какизвлечь максимальное значение в каждой строке, но я не знаю, как получить имя столбца.Я знаю, что в пандах можно использовать idxmax.Но есть ли подобная функция в Numpy?Спасибо!

for i in range(180295):
    lda_trans_max.append(np.max(lda_trans[i]))

Answer 1

Использование np.argmax.

Демо-версия:

>>> a                                                                                                                            
array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])
>>> np.argmax(a, axis=1)                                                                                                             
array([4, 4])

Вы получаете [4, 4] здесь, потому что в обеих строках элемент с максимальным значением находится в позиции 4.

Еще одна демонстрация:

>>> a                                                                                                                                
 array([[5, 9, 7, 6, 8],
       [8, 7, 7, 6, 9]])
>>> np.argmax(a, axis=1)                                                                                                             
array([1, 4])

Answer 2

Numpy довольно математически иногда. Попробуй с Argmax