Получить индекс массива numpy

Question

x=pred_vec.argmax(axis=-1) 
x
o/p-array([51], dtype=int64)

Мне нужно извлечь 51, чтобы с его помощью можно было показать класс, которому он помечен.

Answer 1

Кажется, вы пытаетесь найти индекс максимального значения, используя python и numpy (основываясь на ваших тегах ??) ... Если это так, то попробуйте это:

In [1]: import numpy as np                                                                                                                                                                                                                                        

In [3]: x = np.random.randint(1, 200, (50))                                                                                                                                                                                                                       

In [4]: x                                                                                                                                                                                                                                                         
Out[4]: 
array([186, 117, 150,  68, 118, 197, 169,  48,  90,  67, 117,  90, 133,
       141, 175, 194,  21, 107, 103,  61, 154,  87,  42,  59,  67,  66,
       198,  76, 190,   7, 188, 146, 158,  42, 106, 157,   1, 114, 185,
        58, 169,  89, 120,  81,  28,  49, 197,  73, 169, 142])

In [5]: index_max = np.argmax(x)                                                                                                                                                                                                                                  

In [6]: x[index_max]                                                                                                                                                                                                                                              
Out[6]: 198

Посмотрите на страницу сортировки, поиска и подсчета , чтобы узнать больше вариантов.